Home
last modified time | relevance | path

Searched refs:item_with_index (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/
Dacl_multi_graph_session.cc103 auto item_with_index = AnfAlgo::VisitKernelWithReturnType( in ConstructOutputRef() local
105 auto &anf_node = item_with_index.first; in ConstructOutputRef()
133 auto item_with_index = AnfAlgo::VisitKernelWithReturnType( in ConstructOutputRefByTupleNode() local
136 auto &anf_node = item_with_index.first; in ConstructOutputRefByTupleNode()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc280 for (const auto &item_with_index : output_with_index) { in UpdateRefCountForGraphOutput() local
281 if (!AnfAlgo::OutputAddrExist(item_with_index.first, item_with_index.second, false)) { in UpdateRefCountForGraphOutput()
284 …auto device_address = AnfAlgo::GetMutableOutputAddr(item_with_index.first, item_with_index.second,… in UpdateRefCountForGraphOutput()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc257 auto item_with_index = AnfAlgo::VisitKernelWithReturnType(anf, 0); in CreateNodeOutputTensors() local
258 MS_EXCEPTION_IF_NULL(item_with_index.first); in CreateNodeOutputTensors()
259 MS_LOG(DEBUG) << "Create tensor for output after visit:" << item_with_index.first->DebugString(); in CreateNodeOutputTensors()
261 if (AnfAlgo::CheckPrimitiveType(item_with_index.first, prim::kPrimMakeTuple)) { in CreateNodeOutputTensors()
262 auto cnode = item_with_index.first->cast<CNodePtr>(); in CreateNodeOutputTensors()
272 size_t size = AnfAlgo::GetOutputTensorNum(item_with_index.first); in CreateNodeOutputTensors()
278 const auto &iter = node_to_tensor->find(item_with_index); in CreateNodeOutputTensors()
283 …const auto &tensor = CreateNodeOutputTensor(item_with_index, graph, input_tensors, tensor_to_node); in CreateNodeOutputTensors()
284 (*node_to_tensor)[item_with_index] = tensor; in CreateNodeOutputTensors()
415 auto item_with_index = AnfAlgo::VisitKernelWithReturnType(anf, 0); in CreateNodeOutputPlaceholder() local
[all …]
Danf_runtime_algorithm.cc324 auto item_with_index = AnfAlgo::VisitKernelWithReturnType(node, 0, false, return_prim_type); in GetAllOutput() local
325 if (AnfAlgo::CheckPrimitiveType(item_with_index.first, prim::kPrimMakeTuple)) { in GetAllOutput()
326 MS_EXCEPTION_IF_NULL(item_with_index.first); in GetAllOutput()
327 auto make_tuple = item_with_index.first->cast<CNodePtr>(); in GetAllOutput()
335 ret.push_back(item_with_index.first); in GetAllOutput()
Dkernel_graph.cc54 auto item_with_index = in GetCallRealOutputs() local
56 AnfNodePtr node = item_with_index.first; in GetCallRealOutputs()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc226 for (auto &item_with_index : output_nodes) { in UpdateOutput() local
227 MS_EXCEPTION_IF_NULL(item_with_index.first); in UpdateOutput()
229 if (AnfAlgo::GetOutputTensorNum(item_with_index.first) == 0) { in UpdateOutput()
232 outputs->emplace_back(CreateOutputTensor(item_with_index.first, item_with_index.second)); in UpdateOutput()
238 for (auto &item_with_index : output_nodes) { in UpdateOutputDeviceAddress() local
239 auto &output_node = item_with_index.first; in UpdateOutputDeviceAddress()
240 auto output_index = item_with_index.second; in UpdateOutputDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc259 auto item_with_index = AnfAlgo::VisitKernelWithReturnType(node->input(i), 0); in CreatTensorForOutput() local
260 auto out = CreatTensorForOutput(kernel_graph, item_with_index, tensor_to_node); in CreatTensorForOutput()
300 auto item_with_index = AnfAlgo::VisitKernelWithReturnType(item, 0, false); in CreateOutputTensors() local
301 auto out = CreatTensorForOutput(kernel_graph, item_with_index, tensor_to_node); in CreateOutputTensors()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc650 for (const auto &item_with_index : non_communication_op) { in AssignStaticMemoryOutput() local
651 MS_EXCEPTION_IF_NULL(item_with_index.first); in AssignStaticMemoryOutput()
652 MS_LOG(DEBUG) << "AssignNodeOutputMem for " << item_with_index.first->fullname_with_scope(); in AssignStaticMemoryOutput()
653 AssignNodeOutputMem(kStaticMem, item_with_index.first, SizeToInt(item_with_index.second)); in AssignStaticMemoryOutput()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.cc1192 auto item_with_index = AnfAlgo::VisitKernelWithReturnType(node, 0, true); in IsAllOutGraphOut() local
1193 MS_EXCEPTION_IF_NULL(item_with_index.first); in IsAllOutGraphOut()
1194 if (!item_with_index.first->isa<CNode>() || !AnfAlgo::IsRealKernel(item_with_index.first)) { in IsAllOutGraphOut()
1197 if (item_with_index.first == cnode) { in IsAllOutGraphOut()
1198 output_index_set.insert(item_with_index.second); in IsAllOutGraphOut()