Searched refs:EXPAND_DIMS (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | unsorted_segment_op_info.cc | 284 …auto expandim_output = gen_g.PushBack({gen_g.NewOpInst(EXPAND_DIMS), segment_min, CreatInt64Imm(0)… in ComputeReplaceGraph()
318 …auto expandim_output = gen_g.PushBack({gen_g.NewOpInst(EXPAND_DIMS), segment_max, CreatInt64Imm(0)… in ComputeReplaceGraph()
|
| D | ops_utils.h | 382 constexpr char EXPAND_DIMS[] = "ExpandDims"; variable
|
| D | gather_v2_p_info.cc | 782 auto expand_dims = gen_g.PushBack({gen_g.NewOpInst(EXPAND_DIMS), cast, CreatInt64Imm(axis_ - 1)}); in ComputeReplaceGraph()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 167 …STRIDEDSLICE, GET_NEXT, CAST, NEG, SQUARE, BATCH_MATMUL, EXPAND_DIMS, SQUEEZE, SPARSE_GATHERV2, TI… in IsSplittableOperator()
|
| /third_party/mindspore/third_party/proto/tensorflow/lite/ |
| D | schema.fbs | 295 EXPAND_DIMS = 70,
|
| D | schema_2.4.1.fbs | 297 EXPAND_DIMS = 70,
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | schema.fbs | 295 EXPAND_DIMS = 70,
|