Searched refs:RESHAPE (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | model.py | 91 RESHAPE = 1 variable in PrimLib
113 elem_relation = [PrimLib.RESHAPE] * len(elem_relation)
213 'Reshape': Prim(RESHAPE),
214 'Squeeze': Prim(RESHAPE),
215 'Flatten': Prim(RESHAPE),
216 'FlattenGrad': Prim(RESHAPE),
|
| D | graph_split.py | 237 if self.pattern > PrimLib.RESHAPE:
500 reshape_ops = [op for op in area.ops if PrimLib.iter_type(op) == PrimLib.RESHAPE]
635 if user_area.pattern == PrimLib.RESHAPE:
690 return self.Area.MODE_BASIC if pattern == PrimLib.RESHAPE else self.Area.MODE_COMPOSITE
696 if dom.pattern != PrimLib.RESHAPE:
988 if dom.pattern != PrimLib.RESHAPE:
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 39 Operator op = std::make_pair(RESHAPE, args); in SkipRedisReshapeOP()
59 op_ = std::make_pair(RESHAPE, args); in ReshapeOP()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/ |
| D | op_node.h | 43 RESHAPE, enumerator
161 ReshapeOp(const std::string &op, const std::string &node_name) : PrimOp(op, node_name, RESHAPE) {} in ReshapeOp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 162 …{MATMUL, TRANSPOSE, GELU, TANH, SOFTMAX, SUB, MUL, DIV, RESHAPE, GREATER, LOG_SOFTMAX, ACTIVATION,… in IsSplittableOperator()
261 if (prim->name() == RESHAPE) { in SetStrategyToOperator()
612 bool follow_strategy = (prim->name() == RESHAPE) || (prev_prim->name() == RESHAPE) || in CreateEdgeBetweenTwoOps()
|
| D | step_parallel.cc | 1916 if (prim->name() == RESHAPE) { in ExtractInformation()
2063 if ((node_prim->name() == DEPEND && node_pair.second != 1) || node_prim->name() == RESHAPE) { in FindParameterNextLayout()
2200 if (prim->name() != RESHAPE) { in ReshapeInit()
2842 if (prim->name() != RESHAPE) { in HandleRootReshapeAndSaveStrategy()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_parse_graph.h | 46 {RESHAPE, OperatorType::kRecReshape},
|
| D | rec_generate_strategy.cc | 722 if (ops[incoming_op_index]->type() == RESHAPE || ops[incoming_op_index]->type() == TRANSPOSE) { in PrepareIncomingOperatorInputStrategy()
1183 …type() == REDUCE_SUM || ops[iter_ops]->type() == REDUCE_MEAN || ops[iter_ops]->type() == RESHAPE || in CopyOutgoingOperatorInputStrategy()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.cc | 291 if (prim->name() == RESHAPE) { in FindReshape()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | edge_costmodel.cc | 104 if (edge_name_.find(RESHAPE) != std::string::npos) { in InitEdgeCost()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | ops_utils.h | 270 constexpr char RESHAPE[] = "Reshape"; variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 309 if (prim->name() == RESHAPE) { in CreateOpInfo()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | schema.fbs | 243 RESHAPE = 22,
|
| /third_party/mindspore/third_party/proto/tensorflow/lite/ |
| D | schema.fbs | 243 RESHAPE = 22,
|
| D | schema_2.4.1.fbs | 246 RESHAPE = 22,
|