mindspore2022/mindspore/ccsrc/kernel
zhaozhenlong 562c2b8d3d sparse gather v2 2020-06-11 19:11:56 +08:00
..
aicpu Change uintptr_t to void ptr 2020-05-21 10:15:13 +08:00
akg Change uintptr_t to void ptr 2020-05-21 10:15:13 +08:00
cpu !1992 fix lstm weight initializer 2020-06-11 15:55:57 +08:00
gpu gpu support bert finetune 2020-06-09 11:42:53 +08:00
hccl modify hccl mutiple input and output op regiter 2020-05-29 17:19:43 +08:00
kash initial version 2020-03-27 22:54:54 +08:00
oplib add attr op_pattern to kernel build info 2020-05-14 14:41:27 +08:00
rts enable new label_switch task info 2020-06-10 16:13:14 +08:00
tbe sparse gather v2 2020-06-11 19:11:56 +08:00
CMakeLists.txt add cpu op embedding_look_up 2020-06-05 14:28:18 +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 Add SparseAdam and SparseLazyAdam cpu kernel 2020-06-09 22:24:38 +08:00
common_utils.h Add unique process for duplicated indices 2020-06-09 10:12:39 +08:00
kernel.h convert dropoutdomask nodes's kernel info's first format which connected with same dropoutgenmask 2020-06-08 09:48:34 +08:00
kernel_build_info.cc convert dropoutdomask nodes's kernel info's first format which connected with same dropoutgenmask 2020-06-08 09:48:34 +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 fix bug of tbe adapter 2020-06-09 21:23:35 +08:00
kernel_query.h using device dtype to create transdata kernel build info 2020-05-15 16:10:17 +08:00
task_stream.h initial version 2020-03-27 22:54:54 +08:00