Home
last modified time | relevance | path

Searched defs:kInputSize (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/core/ops/grad/
Dsoft_margin_loss_grad.cc26 constexpr int64_t kInputSize = 3; variable
/third_party/mindspore/mindspore/core/ops/
Dsoft_margin_loss.cc26 constexpr int64_t kInputSize = 2; variable
Dctc_loss_v2_grad.cc30 constexpr int64_t kInputSize = 7; variable
Dctc_loss_v2.cc33 constexpr int64_t kInputSize = 4; variable
Dop_utils.h98 constexpr auto kInputSize = "input_size"; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcheck_valid_cpu_kernel.cc25 constexpr size_t kInputSize = 2; variable
Droi_align_cpu_kernel.cc24 constexpr size_t kInputSize = 2; variable
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_rel_pos_multi_head_attention_fusion.cc33 const size_t kInputSize = 16; variable
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc370 constexpr auto kInputSize = 2; in RemoveNopNode() local