3 Star 14 Fork 13

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

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
ScatterMax.json 1.16 KB
一键复制 编辑 原始数据 按行查看 历史
孙明志 提交于 2024-04-18 21:54 +08:00 . Add ScatterMax
[
{
"op": "ScatterMax",
"language": "cpp",
"input_desc": [
{
"name": "var",
"param_type": "required",
"format": [
"ND", "ND", "ND"
],
"type": [
"fp32", "fp16", "int32"
]
},
{
"name": "indices",
"param_type": "required",
"format": [
"ND", "ND", "ND"
],
"type": [
"int32", "int32", "int32"
]
},
{
"name": "updates",
"param_type": "required",
"format": [
"ND", "ND", "ND"
],
"type": [
"fp32", "fp16", "int32"
]
}
],
"output_desc": [
],
"attr": [
{
"name": "use_locking",
"param_type": "optional",
"type": "bool",
"default_value": false
}
]
}
]
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

搜索帮助