Home
last modified time | relevance | path

Searched defs:kOutputIndex (Results 1 – 25 of 31) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Ddigamma_cpu_kernel.cc32 constexpr size_t kOutputIndex = 0; variable
Dlgamma_cpu_kernel.cc30 constexpr size_t kOutputIndex = 0; variable
Dcrop_and_resize_grad_boxes_cpu_kernel.h51 constexpr size_t kOutputIndex = 0; variable
Dcsr_sparse_matrix_to_dense_cpu_kernel.cc32 constexpr size_t kOutputIndex = 0; variable
Dpolygamma_cpu_kernel.cc33 constexpr size_t kOutputIndex = 0; variable
Ddilation2d_cpu_kernel.cc25 constexpr size_t kOutputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/
Dcholesky_solve_cpu_kernel.cc26 constexpr size_t kOutputIndex = 0; variable
Dcholesky_cpu_kernel.cc30 constexpr size_t kOutputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dapply_proximal_gradient_descent_gpu_kernel.cc32 constexpr size_t kOutputIndex = 0; variable
Dfractional_max_pool_with_fixed_ksize_gpu_kernel.cc26 constexpr size_t kOutputIndex = 0; variable
Dfractional_max_pool_gpu_kernel.cc27 constexpr size_t kOutputIndex = 0; variable
Dfractional_max_pool_grad_with_fixed_ksize_gpu_kernel.cc27 constexpr size_t kOutputIndex = 0; variable
Dfractional_max_pool3d_grad_with_fixed_ksize_gpu_kernel.cc27 constexpr size_t kOutputIndex = 0; variable
Dfractional_max_pool3d_with_fixed_ksize_gpu_kernel.cc26 constexpr size_t kOutputIndex = 0; variable
Dfractional_avg_pool_gpu_kernel.cc27 constexpr size_t kOutputIndex = 0; variable
Dfractional_avg_pool_grad_gpu_kernel.cc28 constexpr size_t kOutputIndex = 0; variable
Dfractional_max_pool_grad_gpu_kernel.cc29 constexpr size_t kOutputIndex = 0; variable
Dsparse_apply_proximal_gradient_descent_gpu_kernel.cc33 constexpr size_t kOutputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dcholesky_gpu_kernel.h39 constexpr size_t kOutputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dbatch_norm_bert_fission.cc31 const std::vector<int64_t> kOutputIndex{0, 3, 4, 5}; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Ddilation2d_helper.h32 constexpr size_t kOutputIndex = 0; variable
Dcrop_and_resize_grad_boxes_helper.h43 constexpr size_t kOutputIndex = 0; variable
Dcrop_and_resize_grad_image_helper.h44 constexpr size_t kOutputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dreverse_sequence.cc24 const int kOutputIndex = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dbatch_norm_add_relu_grad_fusion.cc38 const std::vector<int> kOutputIndex{0, 1, 2}; variable

12