Searched defs:middle_node (Results 1 – 1 of 1) sorted by relevance
390 TensorLayout GetTensorInLayout(const CNodePtr &middle_node, const PrimitivePtr &middle_prim, in GetTensorInLayout()430 … const CNodePtr &middle_node, int64_t index, TensorRedistribution tensor_redistribution, in Redistribution()