Home
last modified time | relevance | path

Searched defs:kInputIndex (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dunique_with_pad_cpu_kernel.h35 inline static constexpr size_t kInputIndex = 0; variable
Ddigamma_cpu_kernel.cc31 constexpr size_t kInputIndex = 0; variable
Dlgamma_cpu_kernel.cc29 constexpr size_t kInputIndex = 0; variable
Ddilation2d_cpu_kernel.cc23 constexpr size_t kInputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Doptimize_updatestate.cc25 constexpr size_t kInputIndex = 1; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dmaxunpool3d_grad_gpu.cc22 constexpr size_t kInputIndex = 0; variable
Dmaxunpool2d_grad_gpu_kernel.cc22 constexpr size_t kInputIndex = 0; variable
Dfractional_max_pool_gpu_kernel.cc24 constexpr size_t kInputIndex = 0; variable
Dfractional_max_pool3d_with_fixed_ksize_gpu_kernel.cc24 constexpr size_t kInputIndex = 0; variable
Dfractional_avg_pool_gpu_kernel.cc24 constexpr size_t kInputIndex = 0; variable
Dfractional_max_pool_with_fixed_ksize_gpu_kernel.cc24 constexpr size_t kInputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dcholesky_gpu_kernel.h37 constexpr size_t kInputIndex = 0; variable
Dcholesky_solve_gpu_kernel.h36 constexpr size_t kInputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dapply_momentum_weight_scale_fusion.cc35 constexpr size_t kInputIndex = 1; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/
Dcholesky_cpu_kernel.cc28 constexpr size_t kInputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Duniform.cc28 const int32_t kInputIndex = 0; variable
Dkldivlossgrad.cc32 const uint32_t kInputIndex = 1; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Ddilation2d_helper.h30 constexpr size_t kInputIndex = 0; variable
Dfractionalpool_helper.h35 constexpr size_t kInputIndex = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dupdatestate_eliminate.cc39 constexpr size_t kInputIndex = 1; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dop_base.h173 #define kInputIndex 0 macro