Home
last modified time | relevance | path

Searched defs:reshape_node (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dexpand_dims_for_batchnorm.cc108 CNodePtr reshape_node = mindspore::common::CreateReshapeNode(graph, input_node, new_shape); in ExpandInputDims() local
132 CNodePtr reshape_node = mindspore::common::CreateReshapeNode(graph, bn_node, bn_output_shape); in ExpandSingleOutputDims() local
157 … CNodePtr reshape_node = mindspore::common::CreateReshapeNode(graph, user, get_item_output_shape); in ExpandMultiOutputDims() local
Dconvert_pad_v3_paddings.cc58 auto reshape_node = NewCNode(reshape_inputs, graph); in CreateReshapeNode() local
356 auto reshape_node = CreateReshapeNode(graph, input_x_node, new_shape); in ExpandInputXDims() local
364 auto reshape_node = CreateReshapeNode(graph, node, output_shape); in ReduceOutputDims() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dshape_reshape_fusion.cc68 void FindFusionPatternNodes(const CNodePtr &reshape_node, std::vector<AnfNodePtr> *fusion_nodes) { in FindFusionPatternNodes()
166 std::shared_ptr<ops::ScalarGraphHolder> CreateScalarGraph(const CNodePtr &reshape_node) { in CreateScalarGraph()
Dinference_matmul_split_fusion.cc79 auto reshape_node = common::AnfAlgo::GetInputNode(cnode, kIndex0); in GetSplitFusionPatternName() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dadjust_resize_dims_pass.cc70 auto reshape_node = func_graph->NewCNode(reshape_prim_c, inputs); in CreateBeforeReshapeNode() local
217 auto reshape_node = func_graph->NewCNode(reshape_prim_c, inputs); in CreateAfterReshapeNode() local
443 auto reshape_node = CreateAfterReshapeNode(func_graph, new_resize_cnode, mul_node); in AdjustResize5DToResize4D() local
Dreshape_reduce_fusion.cc105 auto reshape_node = reduce->input(1); in CheckReshapeReduceFusion() local
Dadjust_matmul_pass.cc185 auto reshape_node = func_graph->NewCNode(reshape_prim_c, inputs); in CreateAfterReshapeNode() local
Dbatchmatmul_fusion.cc218 int ResetReshapeParameters(const AnfNodePtr &reshape_node) { in ResetReshapeParameters()
Dgroupnormsilu_fusion.cc267 auto reshape_node = func_graph->NewCNode(reshape_prim_c, op_inputs); in ReshapeCNode() local
Dflash_attention_fusion.cc132 int32_t GetReshapeParam(const AnfNodePtr &reshape_node, size_t index) { in GetReshapeParam()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc94 auto reshape_node = pass.NewCNode(reshape_inputs, graph); in CreateReshape() local
128 auto reshape_node = CreateReshape(graph, sparse_softmax_node->input(kIndex2), shape, pass); in CreateOneHot() local
183 auto reshape_node = CreateReshape(graph, sparse_softmax_node->input(kIndex1), shape, pass); in CreateSoftmaxCrossEntropyWithLogits() local
652 auto reshape_node = CreateReshape(graph, new_mul_node, logits_shape, *this); in Process() local
694 auto reshape_node = CreateReshape(graph, mul_node, logits_shape, *this); in Process() local
783 auto reshape_node = CreateReshape(graph, new_mul_node, logits_shape, *this); in Process() local
846 auto reshape_node = CreateReshape(graph, new_mul_node, logits_shape, *this); in Process() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/
Dgraph_compiler_test.cc58 auto reshape_node = func_graph->NewCNode(reshape_inputs); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_pass.cc75 auto reshape_node = CreateReshape(new_mul_node, BaseShapeToShape(inputs[kIndex0]->GetShape())); in Run() local
112 auto reshape_node = CreateReshape(inputs[kIndex1], shape); in CreateOneHot() local
138 auto reshape_node = CreateReshape(inputs[kIndex0], shape); in CreateSoftmaxCrossEntropyWithLogits() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/utils/
Danfalgo.h348 auto reshape_node = NewCNode(reshape_inputs, graph); in CreateReshapeNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_pass.cc184 auto reshape_node = CreateReshape(new_mul_node, logits_shape); in Run() local
203 auto reshape_node = ir_bprop_->ad_param()->tape_->FuncGraph::NewCNode(reshape_inputs); in CreateReshape() local
234 auto reshape_node = CreateReshape(sparse_softmax_node->input(kIndex2), shape); in CreateOneHot() local
268 auto reshape_node = CreateReshape(sparse_softmax_node->input(kIndex1), shape); in CreateSoftmaxCrossEntropyWithLogits() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dreshape_ext.cc123 auto reshape_node = graph->GetScalarNode(graph->GetNodeSize() - 1); in ReshapeExtInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dflash_attention_score_info.cc810 auto reshape_node = cnode->input(ops::kFlashAttentionScoreInputDropMaskIndex + 1); in CheckStrategyForDynamicShape() local
1040 auto reshape_node = cnode->input(ops::kFlashAttentionScoreInputDropMaskIndex + 1); in ReplaceNodeInputOrAttrs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc591 auto reshape_node = GetReshape(reshape_cnode); in PreposeWithNormalReshape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_utils.cc1281 Status SegmentEntireShapeToPartialForDynamic(const CNodePtr &reshape_node, const FuncGraphPtr &func… in SegmentEntireShapeToPartialForDynamic()