Home
last modified time | relevance | path

Searched defs:recv_layout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Ddtensor_op_spmd_expander.cc73 const Layout recv_layout = dtensor_recv.layout(); in ValidateSendRecvLayoutConfiguration() local
239 const Layout& recv_layout) { in IsOneToOneHostMeshTransfer()
410 const Layout recv_layout = dtensor_recv.layout(); in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dlower_send_recv.cc98 const Layout recv_layout = in LowerDTensorSendRecvsOps() local
H A Ddtensor_send_recv.cc229 const Layout& recv_layout = dtensor_recv.layout(); in LowerDTensorRecvFromCPUToTFOp() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/tests/
H A Dtensor_layout_test.cc425 StatusOr<Layout> recv_layout = in TEST_F() local
443 StatusOr<Layout> recv_layout = Layout::FromString( in TEST_F() local