Home
last modified time | relevance | path

Searched defs:shape_node (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Drandom_normal_mapper.cc73 auto shape_node = NewCNode(cnode, prim::kPrimShape, {shape_like_node}, {}, kNumberTypeInt32, in Mapper() local
85 …auto shape_node = opt::BuildIntVecParameterNode(func_graph, shape_int32, cnode->fullname_with_scop… in Mapper() local
Dresize_mapper.cc180 …auto shape_node = NewCNode(cnode, prim::kPrimShape, {data_input}, {DIMENSION_4D}, kNumberTypeInt32, in CalResizeShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dadjust_resize_dims_pass.cc59 …AnfNodePtr shape_node = BuildIntVecParameterNode(func_graph, shape, cnode->fullname_with_scope() +… in CreateBeforeReshapeNode() local
131 AnfNodePtr shape_node = in CreateResizeCNodeFor5D() local
175 CNodePtr CreateGatherCNode(const FuncGraphPtr &func_graph, const CNodePtr &shape_node) { in CreateGatherCNode()
207 …terReshapeNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const CNodePtr &shape_node) { in CreateAfterReshapeNode()
355 AnfNodePtr shape_node = in CreateResizeCNodeFor3D() local
Dreshape_reshape_fusion.cc87 auto shape_node = std::make_shared<Var>(); in DefineReshapeReshapePattern() local
104 auto shape_node = reshape_cnode->input(kInputIndexTwo); in GenerateNewShapeParam() local
184 AnfNodePtr shape_node = nullptr; in Process() local
Dadjust_matmul_pass.cc171 …terReshapeNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const CNodePtr &shape_node) { in CreateAfterReshapeNode()
Dflash_attention_fusion.cc2044 …auto shape_node = BuildIntVecParameterNode(func_graph, new_shape, node->fullname_with_scope() + "_… in CreateFlashAttentionNodeForVideoComposer() local
2134 …auto shape_node = BuildIntVecParameterNode(func_graph, new_shape, node->fullname_with_scope() + "_… in CreateFlashAttentionNodeForSD() local
2219 …auto shape_node = BuildIntVecParameterNode(func_graph, new_shape, node->fullname_with_scope() + "_… in CreateFlashAttentionNodeForSDPreMul() local
2330 …auto shape_node = BuildIntVecParameterNode(func_graph, new_shape, node->fullname_with_scope() + "_… in CreateFlashAttentionNodeForSDWithoutCast() local
2585 …auto shape_node = BuildIntVecParameterNode(func_graph, new_shape, node->fullname_with_scope() + "_… in CreateFlashAttentionNodeForSDEinsum() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_deform_conv2d_adjust.cc28 …pNode(const FuncGraphPtr &func_graph, const AnfNodePtr &input_node, const AnfNodePtr &shape_node) { in NewReshapeOpNode()
92 auto shape_node = NewShapeOpNode(func_graph, offset_input); in Adjust() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dshape_reshape_fusion.cc227 for (const auto &shape_node : scalar_graph_holder->GetInputShapeNodes()) { in Process() local
248 auto shape_node = VectorRef({prim::kPrimShape, std::make_shared<Var>()}); in DefinePattern() local
264 auto shape_node = cnode->input(kIndex2); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Drenorm_split.cc92 auto shape_node = opt::CreateValueNodeWithKernelInfo(func_graph, MakeValue(in_shape)); in Process() local
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc92 auto shape_node = CreateTensorInput(kernel_graph, NewValueNode(shape)); in CreateReshape() local
351 auto shape_node = pass.NewCNode(dynamic_shape_inputs, graph); in CreateTile() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Davg_pool_grad_for_ge.cc46 AnfNodePtr shape_node = nullptr; in Process() local
Dbroadcast_for_select.cc75 auto shape_node = opt::CreateValueNodeWithKernelInfo(func_graph, MakeValue(broad_shape)); in AddBroadCastToNode() local
Dshape_unify_mindir.cc133 … ShapeUnifyMindIR::CreateStridedSlice(const FuncGraphPtr &func_graph, const AnfNodePtr &shape_node, in CreateStridedSlice()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dprint_value_type.cc79 auto shape_node = dyn_cast<abstract::Shape>(shape->GetShapeTrack()); in GetOptList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/utils/
Danfalgo.h346 auto shape_node = CreateShapeVectorNode(shape); in CreateReshapeNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dbroadcast_for_select.cc77 auto shape_node = NewValueNode(shape_value); in AddBroadCastToNode() local
Dscalar_op_pass.cc231 …tr ScalarOpPass::GenerateStridedSlice(const FuncGraphPtr &func_graph, const AnfNodePtr &shape_node, in GenerateStridedSlice()
Dslice_prepose_pass.cc397 auto shape_node = BuildIntVecParameterNode( in CreateReshapeCNode() local
599 auto shape_node = BuildIntVecParameterNode( in PreposeWithNormalReshape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dcdist_fission.cc78 auto shape_node = opt::CreateValueNodeWithKernelInfo(func_graph, MakeValue(need_shape)); in AddBroadCastToNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_pass.cc90 …auto shape_node = func_builder_->NewFuncNode(PyNativeAlgo::Common::CreateTensorByConstantValue(Mak… in CreateReshape() local
167 …auto shape_node = func_builder_->EmitOp(func_builder_->NewPrimitive("DynamicShape"), {inputs[kInde… in CreateTile() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dinsert_format_transform_op.cc119 auto shape_node = CreateTupleValue(kernel_graph, reshape_shape); in InsertTransposeOp() local
Dprint_reduce_fusion.cc91 auto shape_node = dyn_cast<abstract::Shape>(shape->GetShapeTrack()); in GetOptList() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/optimizer/
Dtensorrt_optimizer.cc144 auto shape_node = slice_input->cast<CNodePtr>(); in GetMatmulFactor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dinplace_assign_builder.cc158 …auto shape_node = CreateTensorValueNode({kOpFormat_DEFAULT, {SizeToLong(device_shape.size())}, kIn… in CreateCleanCompositeNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_pass.cc200 auto shape_node = NewValueNode(shape); in CreateReshape() local
315 auto shape_node = ir_bprop_->ad_param()->tape_->FuncGraph::NewCNode(dynamic_shape_inputs); in CreateTile() local

12