mindspore2022/mindspore/ccsrc/kernel
mindspore-ci-bot ac3c35c329 !2749 GPU update argmaxwithvalue
Merge pull request !2749 from VectorSL/argmaxwithvalue
2020-07-04 10:12:46 +08:00
..
aicpu Init GraphKernel. 2020-06-20 22:31:54 +08:00
akg Add test case and fix two bugs 2020-07-02 12:52:15 +08:00
cpu Adjust the thread num to compute optimizer in cpu 2020-06-30 19:25:34 +08:00
gpu !2749 GPU update argmaxwithvalue 2020-07-04 10:12:46 +08:00
hccl fix_hccl_to_support_big_tensor 2020-06-23 21:37:23 +08:00
kash code review stage 2 2020-06-23 18:36:12 +08:00
oplib code review stage 2 2020-06-23 18:36:12 +08:00
rts fix switch input type 2020-06-28 19:19:49 +08:00
tbe Add TBE operators SparseApplyFtrlV2\SparseApplyAdagradV2 for VM. 2020-07-02 12:00:58 +08:00
CMakeLists.txt Init GraphKernel. 2020-06-20 22:31:54 +08:00
ascend_kernel_mod.h save graph and task desc info to framework file 2020-05-11 09:41:40 +08:00
common_utils.cc Adjust the thread num to compute optimizer in cpu 2020-06-30 19:25:34 +08:00
common_utils.h Adjust the thread num to compute optimizer in cpu 2020-06-30 19:25:34 +08:00
kernel.h Init GraphKernel. 2020-06-20 22:31:54 +08:00
kernel_build_info.cc FormatAgnostic without checking format in kernel build info 2020-06-15 10:03:18 +08:00
kernel_build_info.h fix bug of tbe adapter 2020-06-09 21:23:35 +08:00
kernel_fusion.cc add pre compile process 2020-06-11 11:51:22 +08:00
kernel_fusion.h initial version 2020-03-27 22:54:54 +08:00
kernel_query.cc GraphKernel support akg batchmatmul 2020-06-26 13:52:45 +08:00
kernel_query.h Init GraphKernel. 2020-06-20 22:31:54 +08:00
task_stream.h initial version 2020-03-27 22:54:54 +08:00