Home
last modified time | relevance | path

Searched refs:OUTPUT_NUM (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Drandom_choice_with_mask_cpu_kernel.h28 constexpr size_t OUTPUT_NUM = 2; variable
Droi_align_grad_cpu_kernel.h28 constexpr size_t OUTPUT_NUM = 1; variable
Dnms_with_mask_cpu_kernel.h32 constexpr size_t OUTPUT_NUM = 3; variable
Dcrop_and_resize_cpu_kernel.h33 constexpr size_t OUTPUT_NUM = 1; variable
Drandom_choice_with_mask_cpu_kernel.cc87 if (output_num != OUTPUT_NUM) { in InitKernel()
Droi_align_grad_cpu_kernel.cc72 if (output_num != OUTPUT_NUM) { in CheckParam()
Dcrop_and_resize_cpu_kernel.cc31 if (output_num != OUTPUT_NUM) { in InitKernel()
Dnms_with_mask_cpu_kernel.cc204 if (output_num != OUTPUT_NUM) { in InitKernel()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dsplit_info.cc59 auto output_num_iter = attrs_.find(OUTPUT_NUM); in GetAttrs()
Dops_utils.h160 constexpr char OUTPUT_NUM[] = "output_num"; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dconstruct_operator.cc188 Attr attr_split = std::make_pair(OUTPUT_NUM, attr_value_split); in SplitOP()