Home
last modified time | relevance | path

Searched defs:OUTPUT_INDEX (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/custom/aot_test_files/
Dadd.cc31 constexpr int OUTPUT_INDEX = 2; in CustomAdd() local
Dsquare.cu29 constexpr int OUTPUT_INDEX = 1; in CustomSquare() local
Dhetero_square_mul.cu31 constexpr int OUTPUT_INDEX = 2; in CustomHSquareMul() local
Dreorganize.cu31 constexpr int OUTPUT_INDEX = 2; in CustomReorganize() local
Dsquare_bprop.cu30 constexpr int OUTPUT_INDEX = 3; in CustomSquareBprop() local
Dadd_mul_div.cu33 constexpr int OUTPUT_INDEX = 2; in CustomAddMulDiv() local
Dadd.cu40 constexpr int OUTPUT_INDEX = 2; in CustomAdd() local
Dadd_mul_div_bprop.cu32 constexpr int OUTPUT_INDEX = 6; in CustomAddMulDivBprop() local
Dadd_with_attr.cc51 constexpr int OUTPUT_INDEX = 2; in CustomAdd() local
Dadd_with_attr.cu63 constexpr int OUTPUT_INDEX = 2; in CustomAdd() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/op_declare/
Dop_declare_macro.h98 #define OUTPUT_INDEX(x) \ macro
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dop_base.h122 #define OUTPUT_INDEX 0 macro