Searched refs:kRealInputIndexInDepend (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | recompute_prepare.h | 64 GetRealOutputNodes(output_cnode->input(kRealInputIndexInDepend), real_outputs); in GetRealOutputNodes()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | segment_runner.cc | 126 args.emplace_back(RefSubGraphNode(fg, inps[kRealInputIndexInDepend], &inputs, &eqv)); in TransformSegmentToAnfGraph()
|
| D | backend.cc | 893 …ConstructOutputs(depend_node->input(kRealInputIndexInDepend), output_tensors, output_position, out… in ConstructOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | recompute.cc | 271 …mitiveCNode(node, prim::kPrimDepend) && WithRecomputedScope(node->input(kRealInputIndexInDepend))); in SetRecomputedScope()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 108 …return GetRealOutputRecursively(node->cast<CNodePtr>()->input(kRealInputIndexInDepend), 0, inputs); in GetRealOutputRecursively()
263 return VisitKernel(cnode->input(kRealInputIndexInDepend), 0); in VisitKernel()
307 …return VisitKernelWithReturnType(cnode->input(kRealInputIndexInDepend), index, skip_nop_node, retu… in VisitKernelWithReturnType()
358 auto real_output = GetAllOutputWithIndex(depend_node->input(kRealInputIndexInDepend)); in GetAllOutputWithIndex()
2213 GetAllFatherRealNode(cnode->input(kRealInputIndexInDepend), result, visited); in GetAllFatherRealNode()
|
| D | session_basic.cc | 700 } else if ((is_depend && input_idx > kRealInputIndexInDepend)) { in GetNewCNodeInputs()
1959 return GetSupportedInternalNode(inputs[kRealInputIndexInDepend]); in GetSupportedInternalNode()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 526 constexpr auto kRealInputIndexInDepend = 1; variable
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_scheduler.cc | 1210 …if (AnfAlgo::IsOneOfPrimitiveCNode(input_cnode->input(kRealInputIndexInDepend), recursion_prims)) { in LinkControlArrowByAutoMonad()
1211 real_depend_inputs.push_back(input_cnode->input(kRealInputIndexInDepend)); in LinkControlArrowByAutoMonad()
|