Home
last modified time | relevance | path

Searched defs:OUTPUT_NUM (Results 1 – 25 of 75) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dnms_with_mask_cpu_kernel.h32 constexpr size_t OUTPUT_NUM = 3; variable
Dcrop_and_resize_cpu_kernel.h34 constexpr size_t OUTPUT_NUM = 1; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dsoftsign_helper.h38 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dmish_helper.h38 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dpopulation_count_helper.h39 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Derfinv_helper.h39 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dmaxunpool3d_helper.h50 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
150 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dmaxunpool2d_helper.h50 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
156 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dhsvtorgb_helper.h38 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Drgbtohsv_helper.h38 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dunravel_index_helper.h40 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dlogit_helper.h49 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dlogit_grad_helper.h49 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Daddcdiv_helper.h44 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dunique_helper.h27 constexpr size_t OUTPUT_NUM = 1; variable
Dadaptive_avg_pool3d_helper.h45 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dargmin_helper.h47 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dsoftmarginloss_helper.h47 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dbucketize_helper.h47 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dadaptive_avg_pool3d_grad_helper.h38 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Dheaviside_helper.h40 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
Daddcmul_helper.h44 constexpr size_t OUTPUT_NUM = 1; in CalMemSize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dadjust_saturation_gpu_kernel.cc27 constexpr size_t OUTPUT_NUM = 1; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dassign_sub_gpu_kernel.cc27 constexpr int OUTPUT_NUM = 1; in Init() local
Dassign_add_gpu_kernel.cc28 constexpr int OUTPUT_NUM = 1; in Init() local

123