Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Ddtensor_op_spmd_expander.cc76 const Mesh& recv_mesh = recv_layout.mesh(); in ValidateSendRecvLayoutConfiguration() local
123 bool SendRecvOpUsesXla(const Mesh& send_mesh, const Mesh& recv_mesh) { in SendRecvOpUsesXla()
241 const Mesh& recv_mesh = recv_layout.mesh(); in IsOneToOneHostMeshTransfer() local
411 const Mesh& recv_mesh = recv_layout.mesh(); in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dlower_send_recv.cc81 const Mesh recv_mesh = dtensor_recv.layout().mesh(); in LowerDTensorSendRecvsOps() local