3 Star 14 Fork 12

孙明志/昇腾AI原生创新算子挑战赛S1-0xCCCCCCCC

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
MseLossGrad.json 1.34 KB
一键复制 编辑 原始数据 按行查看 历史
孙明志 提交于 2024-04-16 21:07 +08:00 . Update
[
{
"op": "MseLossGrad",
"language": "cpp",
"input_desc": [
{
"name": "predict",
"param_type": "required",
"format": [
"ND", "ND"
],
"type": [
"fp32", "fp16"
]
},
{
"name": "label",
"param_type": "required",
"format": [
"ND", "ND"
],
"type": [
"fp32", "fp16"
]
},
{
"name": "dout",
"param_type": "required",
"format": [
"ND", "ND"
],
"type": [
"fp32", "fp16"
]
}
],
"output_desc": [
{
"name": "y",
"param_type": "required",
"format": [
"ND", "ND"
],
"type": [
"fp32", "fp16"
]
}
],
"attr": [
{
"name": "value",
"param_type": "optional",
"type": "string",
"default_value": "mean"
}
]
}
]
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/sunkafei/ascend-operator-challenge.git
git@gitee.com:sunkafei/ascend-operator-challenge.git
sunkafei
ascend-operator-challenge
昇腾AI原生创新算子挑战赛S1-0xCCCCCCCC
master

搜索帮助