Home
last modified time | relevance | path

Searched refs:GetTensorInLayout (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.h63 TensorLayout GetTensorInLayout(const CNodePtr &pre_node, const PrimitivePtr &pre_prim,
Dstep_parallel.cc390 TensorLayout GetTensorInLayout(const CNodePtr &middle_node, const PrimitivePtr &middle_prim, in GetTensorInLayout() function
464 TensorLayout tensorlayout_in = GetTensorInLayout(middle_node, middle_prim, distribute_operator); in Redistribution()
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc491 TEST_F(TestStepParallel, GetTensorInLayout) { in TEST_F() argument
519 TensorLayout tensorlayout = GetTensorInLayout(node1, prim, distribute_operator_pre); in TEST_F()