Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbatch_norm_bert_fission.cc27 const std::vector<int64_t> kOutputIndex{0, 3, 4, 5}; variable
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_add_relu_grad_fusion.cc32 const std::vector<int> kOutputIndex{0, 1, 2}; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dop_base.h93 #define kOutputIndex 0 macro