Home
last modified time | relevance | path

Searched defs:send_layout (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Ddtensor_op_spmd_expander.cc72 const Layout& send_layout = send_layout_or_null.value(); in ValidateSendRecvLayoutConfiguration() local
238 bool IsOneToOneHostMeshTransfer(const Layout& send_layout, in IsOneToOneHostMeshTransfer()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/tests/
H A Dtensor_layout_test.cc421 StatusOr<Layout> send_layout = in TEST_F() local
439 StatusOr<Layout> send_layout = Layout::FromString( in TEST_F() local