| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | masked_select_cpu_kernel.cc | 36 node_wpt_ = kernel_node; in InitKernel()
65 if (!node_wpt_.expired()) { in Launch()
66 auto node_ = node_wpt_.lock(); in Launch()
|
| D | unique_cpu_kernel.cc | 25 node_wpt_ = kernel_node; in InitKernel()
56 if (!node_wpt_.expired()) { in Launch()
57 auto node_ = node_wpt_.lock(); in Launch()
|
| D | reshape_cpu_kernel.cc | 29 node_wpt_ = kernel_node; in InitKernel()
46 auto node = node_wpt_.lock(); in Launch()
|
| D | embedding_look_up_cpu_kernel.cc | 59 node_wpt_ = kernel_node; in InitKernel()
84 if (!node_wpt_.expired()) { in LaunchKernel()
85 auto node = node_wpt_.lock(); in LaunchKernel()
|
| D | map_uniform_cpu_kernel.cc | 33 node_wpt_ = kernel_node; in InitKernel()
54 auto node = node_wpt_.lock(); in LaunchKernel()
|
| D | concat_cpu_kernel.cc | 30 node_wpt_ = kernel_node; in InitKernel()
41 auto node = node_wpt_.lock(); in Launch()
|
| D | sub_and_filter_cpu_kernel.cc | 31 node_wpt_ = kernel_node; in InitKernel()
53 auto node = node_wpt_.lock(); in LaunchKernel()
|
| D | update_cache_cpu_kernel.cc | 33 node_wpt_ = kernel_node; in InitKernel()
63 auto node = node_wpt_.lock(); in LaunchKernel()
|
| D | dynamic_assign_cpu_kernel.cc | 33 node_wpt_ = kernel_node; in InitKernel()
61 auto node = node_wpt_.lock(); in LaunchKernel()
|
| D | pad_and_shift_cpu_kernel.cc | 29 node_wpt_ = kernel_node; in InitKernel()
84 auto node_ = node_wpt_.lock(); in LaunchKernel()
|
| D | map_uniform_cpu_kernel.h | 45 CNodeWeakPtr node_wpt_; variable
|
| D | sub_and_filter_cpu_kernel.h | 44 CNodeWeakPtr node_wpt_; variable
|
| D | pad_and_shift_cpu_kernel.h | 45 CNodeWeakPtr node_wpt_; variable
|
| D | concat_cpu_kernel.h | 41 CNodeWeakPtr node_wpt_; variable
|
| D | dynamic_assign_cpu_kernel.h | 46 CNodeWeakPtr node_wpt_; variable
|
| D | embedding_look_up_cpu_kernel.h | 46 CNodeWeakPtr node_wpt_; variable
|
| D | map_cache_idx_cpu_kernel.cc | 66 node_wpt_ = kernel_node; in InitKernel()
96 auto node = node_wpt_.lock(); in LaunchKernel()
|
| D | masked_select_cpu_kernel.h | 43 CNodeWeakPtr node_wpt_; variable
|
| D | map_cache_idx_cpu_kernel.h | 46 CNodeWeakPtr node_wpt_; variable
|
| D | update_cache_cpu_kernel.h | 49 CNodeWeakPtr node_wpt_; variable
|
| D | reshape_cpu_kernel.h | 38 CNodeWeakPtr node_wpt_;
|
| D | unique_cpu_kernel.h | 64 CNodeWeakPtr node_wpt_; variable
|