Home
last modified time | relevance | path

Searched defs:value_node (Results 1 – 25 of 286) sorted by relevance

12345678910>>...12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dprimitive_adjust.cc215 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrMapCommon() local
232 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrMapArgMaxWithValue() local
250 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrMapActivation() local
272 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrMapActivationGrad() local
294 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrMapReduce() local
339 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrMapConv2D() local
379 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrPool() local
417 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrPoolGrad() local
455 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrMapAdder() local
477 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in MoveAttrMapLayerNorm() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dgraph_adapter.cc45 auto value_node = node->cast<ValueNodePtr>(); in GetTensorFromValueNode() local
75 auto value_node = input->cast<ValueNodePtr>(); in GetGraphValueNodeRefCounts() local
86 auto value_node = output->cast<ValueNodePtr>(); in GetGraphValueNodeRefCounts() local
95 device::DeviceAddressPtr CreateValueNodeAddress(const ValueNodePtr &value_node, in CreateValueNodeAddress()
148 const ValueNodePtr &value_node, in HandleAddressForHeterogeneous()
178 for (const auto &value_node : graph->graph_value_nodes()) { in RemoveUnusedValueNodes() local
191 for (auto &value_node : graph->graph_value_nodes()) { in ClearForwardOutputValueNodeDeviceAddress() local
222 for (auto &value_node : graph->graph_value_nodes()) { in GenerateRefCountForBpropValueNode() local
267 void GraphAdapter::HandleBackoffValueNode(const ValueNodePtr &value_node, const AnfNodePtr &front_n… in HandleBackoffValueNode()
322 for (auto &value_node : graph->graph_value_nodes()) { in UpdateForwardOutputInBpropGraph() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/
Danfnode_tensor_adapter.cc61 InferTensor *TensorAdapter::Convert2Tensor(const ValueNodePtr &value_node, Format format) { in Convert2Tensor()
208 TensorAdapterPtr TensorAdapter::CreateFromTensorValueNode(const ValueNodePtr &value_node) { in CreateFromTensorValueNode()
238 TensorAdapterPtr TensorAdapter::CreateFromInt32ImmValue(const ValueNodePtr &value_node) { in CreateFromInt32ImmValue()
261 TensorAdapterPtr TensorAdapter::CreateFromInt64ImmValue(const ValueNodePtr &value_node) { in CreateFromInt64ImmValue()
284 TensorAdapterPtr TensorAdapter::CreateFromBoolImmValue(const ValueNodePtr &value_node) { in CreateFromBoolImmValue()
312 TensorAdapterPtr TensorAdapter::CreateFromNumberTypeValue(const ValueNodePtr &value_node) { in CreateFromNumberTypeValue()
341 TensorAdapterPtr TensorAdapter::CreateFromIntSequenceValue(const ValueNodePtr &value_node) { in CreateFromIntSequenceValue()
407 TensorAdapterPtr TensorAdapter::Create(const ValueNodePtr &value_node, Format format) { in Create()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dseed_adapter.cc60 auto value_node = std::make_shared<ValueNode>(tensor); in CreateValueNode() local
96 auto value_node = CreateValueNode(attr_value); in ConvertAttrToValueNode() local
140 for (auto &value_node : value_nodes) { in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dmatmul_fusion_mapper.cc77 ValueNodePtr value_node = nullptr; in Mapper() local
90 ValueNodePtr value_node = nullptr; in Mapper() local
153 ValueNodePtr value_node = nullptr; in QuantMapper() local
Dsqueeze_mapper.cc35 auto value_node = input_node->cast<ValueNodePtr>(); in GetAxisValue() local
85 ValueNodePtr value_node = nullptr; in Mapper() local
Dtopk_fusion_mapper.cc70 ValueNodePtr value_node = nullptr; in Mapper() local
100 ValueNodePtr value_node = nullptr; in Mapper() local
Dfused_batchnorm_mapper.cc24 ValueNodePtr value_node = nullptr; in Mapper() local
Dconv3d_mapper.cc25 ValueNodePtr value_node = nullptr; in Mapper() local
Ddepth_to_space_mapper.cc26 ValueNodePtr value_node = nullptr; in Mapper() local
Dspace_to_depth_mapper.cc30 ValueNodePtr value_node = nullptr; in Mapper() local
Dargmin_fusion_mapper.cc31 ValueNodePtr value_node = nullptr; in Mapper() local
Dlayernorm_fusion_mapper.cc28 ValueNodePtr value_node = nullptr; in Mapper() local
Dstridedslice_mapper.cc28 ValueNodePtr value_node = nullptr; in Mapper() local
Dsilu_mapper.cc39 ValueNodePtr value_node = nullptr; in Mapper() local
Dbatchnorm_mapper.cc28 ValueNodePtr value_node = nullptr; in Mapper() local
Dconv2d_backprop_input_mapper.cc32 ValueNodePtr value_node = nullptr; in Mapper() local
Dconcatv2_mapper.cc32 ValueNodePtr value_node = nullptr; in Mapper() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgenerate_graph.cc67 auto value_node = inputs[0]->cast<ValueNodePtr>(); in RectifyInputsForNewCNode() local
255 auto value_node = NewValueNode(value_ptr); in ValuePtrToAnfNodePtr() local
346 auto value_node = NewValueNode(op); in InsertVirtualPipelineEndNode() local
364 auto value_node = NewValueNode(op); in CreateVirtualConverterBeginNode() local
376 auto value_node = NewValueNode(op); in CreateVirtualConverterEndNode() local
433 auto value_node = NewValueNode(op_prim_instance); in NewOpInst() local
445 auto value_node = NewValueNode(op_prim_instance); in NewOpInst() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_dtype_adjust.cc31 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in GetPrimFromCnode() local
120 ValueNodePtr value_node = nullptr; in Adjust() local
Donnx_megatron_op_adjust.cc37 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in AdjustMegatronAllReduce() local
74 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in AdjustMegatronLinearAllGather() local
109 auto value_node = cnode->input(0)->cast<ValueNodePtr>(); in AdjustMegatronScaledMaskedSoftmax() local
Donnx_deform_conv2d_adjust.cc37 ValueNodePtr value_node = NewValueNode(prim_c); in NewReshapeOpNode() local
54 ValueNodePtr value_node = NewValueNode(prim_c); in NewShapeOpNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dvalue_depend_op_utils.cc68 auto value_node = input_node->cast<ValueNodePtr>(); in IsConstInput() local
115 auto value_node = input_node->cast<ValueNodePtr>(); in AddConstInputToAttr() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dspecial_op_eliminate.h401 ValueNodePtr value_node = node->cast<ValueNodePtr>(); in CheckNameSpaceVNode() local
410 ValueNodePtr value_node = node->cast<ValueNodePtr>(); in CheckSymbolVNode() local
417 ValueNodePtr value_node = node->cast<ValueNodePtr>(); in CheckStrVNode() local
477 auto value_node = rep->cast<ValueNodePtr>(); in OperatorHandle1() local
491 auto value_node = rep->cast<ValueNodePtr>(); in OperatorHandle2() local
506 auto value_node = rep->cast<ValueNodePtr>(); in OperatorHandle3() local
528 auto value_node = rep->cast<ValueNodePtr>(); in OperatorHandle4() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/lite_exporter/
Dfetch_content.cc128 int FetchFromTensorValue(const ValueNodePtr &value_node, converter::FmkType fmk_type, bool train_fl… in FetchFromTensorValue()
170 int FetchCastImmValue(const ValueNodePtr &value_node, DataInfo *data_info) { in FetchCastImmValue()
190 int FetchImmValue(const ValueNodePtr &value_node, DataInfo *data_info) { in FetchImmValue()
206 int FetchFromNumberValue(const ValueNodePtr &value_node, DataInfo *data_info) { in FetchFromNumberValue()
224 int FetchFromSequenceValue(const ValueNodePtr &value_node, DataInfo *data_info) { in FetchFromSequenceValue()
364 auto value_node = cnode->input(index)->cast<ValueNodePtr>(); in FetchDataFromValueNode() local

12345678910>>...12