Home
last modified time | relevance | path

Searched defs:IsNeedUpdateOutputShapeAndSize (Results 1 – 25 of 65) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dlist_diff_cpu_kernel.h43 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dnon_max_suppression_with_overlaps_cpu_kernel.h41 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dcsr_sparse_matrix_to_dense_cpu_kernel.h45 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dcoalesce_cpu_kernel.h49 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Ddynamic_assign_cpu_kernel.h43 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dsub_and_filter_cpu_kernel.h58 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dsparse_slice_cpu_kernel.h47 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dpad_and_shift_cpu_kernel.h45 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dsparse_matrix_add_cpu_kernel.h40 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/
Dquant_batch_matmul_aclnn_kernel.h33 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dunique_dim_aclnn_kernel.h34 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dmasked_select_aclnn_kernel.h34 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dunique2_aclnn_kernel.h34 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dnon_zero_aclnn_kernel.h34 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dunique_with_pad_gpu_kernel.h41 bool IsNeedUpdateOutputShapeAndSize() override { return false; } in IsNeedUpdateOutputShapeAndSize() function
Dcoalesce_gpu_kernel.h44 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dunique_consecutive_gpu_kernel.h44 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dnonmaxsuppressionv3_gpu_kernel.h45 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/acl_kernel/
Dgetnext_kernel_mod.h32 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
Dcustom_op_kernel_mod.h36 bool IsNeedUpdateOutputShapeAndSize() override { in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/data/
Ddataset_iterator_kernel.h41 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/
Dtensor_array_stack_kernel.h39 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse_grad/
Dsparse_add_grad_gpu_kernel.h51 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/
Dhcom_receive.h44 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/rl/
Dtensor_array_stack_kernel.h37 bool IsNeedUpdateOutputShapeAndSize() override { return true; } in IsNeedUpdateOutputShapeAndSize() function

123