Searched refs:middle_prim (Results 1 – 1 of 1) sorted by relevance
390 TensorLayout GetTensorInLayout(const CNodePtr &middle_node, const PrimitivePtr &middle_prim, in GetTensorInLayout() argument393 if (middle_prim->name() == prim::kTupleGetItem) { in GetTensorInLayout()440 PrimitivePtr middle_prim = middle_value->value()->cast<PrimitivePtr>(); in Redistribution() local441 MS_EXCEPTION_IF_NULL(middle_prim); in Redistribution()448 …MS_LOG(DEBUG) << "Redistribution: middle_prim " << middle_prim->name() << " next_prim " << next_pr… in Redistribution()464 TensorLayout tensorlayout_in = GetTensorInLayout(middle_node, middle_prim, distribute_operator); in Redistribution()469 …MS_LOG(ERROR) << "Redistribution: middle_prim " << middle_prim->name() << " next_prim : " << next_… in Redistribution()