Searched refs:middle_node (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 69 … const CNodePtr &middle_node, int64_t index, TensorRedistribution tensor_redistribution,
|
| D | step_parallel.cc | 390 TensorLayout GetTensorInLayout(const CNodePtr &middle_node, const PrimitivePtr &middle_prim, in GetTensorInLayout() argument
394 auto value_node = middle_node->input(2)->cast<ValueNodePtr>(); in GetTensorInLayout()
430 … const CNodePtr &middle_node, int64_t index, TensorRedistribution tensor_redistribution, in Redistribution() argument
432 FuncGraphPtr func_graph = middle_node->func_graph(); in Redistribution()
438 auto middle_value = middle_node->input(0)->cast<ValueNodePtr>(); in Redistribution()
449 …MS_LOG(DEBUG) << "Redistribution: middle_node " << middle_node->ToString() << " next_node " << nex… in Redistribution()
464 TensorLayout tensorlayout_in = GetTensorInLayout(middle_node, middle_prim, distribute_operator); in Redistribution()
465 if (IsPrimitiveCNode(middle_node, prim::kPrimReceive)) { in Redistribution()
466 tensorlayout_in = *(middle_node->user_data<TensorLayout>()); in Redistribution()
470 MS_LOG(ERROR) << "Redistribution: middle_node " << middle_node->ToString() << " next_node " in Redistribution()
|