Home
last modified time | relevance | path

Searched defs:node_with_index (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Doutput_actor.cc241 auto node_with_index = input_data->data_->GetNodeIndex(); in RunOpData() local
396 auto node_with_index = device_tensor->GetNodeIndex(); in UpdateOutputDeviceAddress() local
Dany_type_kernel_actor.cc600 …for (const auto &node_with_index : common::AnfAlgo::GetAllOutputWithOutMonadAndParameter(graph()->… in Init() local
723 auto node_with_index = graph_output_data->data_->node_index(); in FetchGraphOutput() local
Ddata_source_actor.cc229 for (auto &node_with_index : data_node_with_indexs_) { in FillDataBuffer() local
Ddata_prepare_actor.cc959 void DataPrepareActor::PrepareDataForControlValueNode(const KernelWithIndex &node_with_index, in PrepareDataForControlValueNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dreduce_same_op_in_horizon.cc111 std::map<AnfNodePtr, size_t> node_with_index; in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/
Dexit_actor.cc344 const KernelWithIndex &node_with_index = input_device_tensor->GetNodeIndex(); in CopyDeviceAddress() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dcontrol_node_parser.cc38 bool IsFrontValueNode(const KernelWithIndex &node_with_index) { in IsFrontValueNode()
49 KernelWithIndex FetchRealInputNode(const KernelWithIndex &node_with_index) { in FetchRealInputNode()
137 auto node_with_index = common::AnfAlgo::VisitKernelWithReturnType(node.first, node.second); in FetchRealParameterByNode() local
544 const auto &node_with_index = in FetchInputNodeByNode() local
766 KernelWithIndex FetchRealNodeByGetItem(const KernelWithIndex &node_with_index) { in FetchRealNodeByGetItem()
925 …const auto &node_with_index = common::AnfAlgo::VisitKernelWithReturnType(node, 0, false, return_ty… in FetchRealDependNodeByAutoMonad() local
1878 …lWithIndex ControlNodeParser::FetchBackendNodeByFrontNode(const KernelWithIndex &node_with_index) { in FetchBackendNodeByFrontNode()
Dcontrol_node_scheduler.cc201 const auto &node_with_index = node_with_index_with_context.first; in BuildDataSourceActorForControlNode() local
Dgraph_scheduler.cc1274 auto node_with_index = graph->GetElementInTupleBackendFrontIndexMap(input_node); in BuildDataSourceActor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dir_converter.cc213 device::DeviceAddressPtr origin_address, session::KernelWithIndex node_with_index) in Edge()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_graph_executor.cc270 const auto &node_with_index = common::AnfAlgo::FetchRealNodeSkipMonadControl(output); in EnableGraphOutputZeroCopy() local
1033 for (auto &node_with_index : graph_outputs_all) { in AddRefCorrespondPairs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dtransform_op_optimizer.cc634 size_t GetNodeId(const NodeWithIndex &node_with_index) { in GetNodeId()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Danfalgo.cc391 KernelWithIndex AnfAlgo::FetchRealNodeSkipMonadControl(const KernelWithIndex &node_with_index) { in FetchRealNodeSkipMonadControl()
459 for (const auto &node_with_index : graph_outputs) { in GetAllOutputWithOutMonadAndParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc1049 …const AnfWithOutIndex &node_with_index, void *const device_ptr, size_t size, const string &format,… in CreateOutputKernelTensorWithDeviceInfo()