在此一键设置昵称,即可参与社区互动!
BubtLTK
发帖: 0粉丝: 0
发消息 + 关注
十分感谢您的帮助!
【功能模块】
argmin:计算输入Tensor指定维度的最小值的索引号
【操作步骤&问题现象】
1、通过原生TensorFlow的算子接口生成测试的期望数据。
2、ut测试代码可以成功读入期望数据,但执行算子后,输出的output还是初始化时的数据
执行算子前的output如下:
执行算子后的output如下:
算子实例化过程如下:
ut测试反馈如下(最后一行64:Failure定位到的是代码RUN_KERNEL):
分享文章到朋友圈
莫晓忧
发帖: 0粉丝: 1
级别 : 版主
我使用你的代码本地调试后发现你的ut下的cmakelists内的set(_cpu_kernels_src下没有添加ArgMin.cc,只需要添加一下源码文件路径就能编译通过了,如图:
请问有在ops/built-in/aicpu/impl/CMakeLists.txt内配置编译算子实现文件么?
ops/built-in/aicpu/impl/CMakeLists.txt内已经配置了编译算子实现文件,下图中的arg_min.cc文件就是,已配置在代码段set(local_kernels_src_files......)中
算子信息库内配置算子信息了么?
算子信息库已配置算子信息,信息如下:
方便把ArgMin相关的修改文件发我一下么?我这边编译定位一下。
实现代码里有写REGISTER_CPU_KERNEL(argmin, ArgMinCpuKernel);去注册该算子实现么?
评论 莫晓忧:嗯,算子实现代码已注册。
zhujianpeng
发帖: 16粉丝: 1
级别 : 版主,版块专家
mark
游客
回复主题
您发表的内容存在敏感词汇
如点击继续发布,文章中的敏感词将以*代替。
方便把ArgMin相关的修改文件发我一下么?我这边编译定位一下。
... 查看全部