Home
last modified time | relevance | path

Searched defs:send_mesh (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Ddtensor_op_spmd_expander.cc75 const Mesh& send_mesh = send_layout.mesh(); in ValidateSendRecvLayoutConfiguration() local
123 bool SendRecvOpUsesXla(const Mesh& send_mesh, const Mesh& recv_mesh) { in SendRecvOpUsesXla()
240 const Mesh& send_mesh = send_layout.mesh(); in IsOneToOneHostMeshTransfer() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dlower_send_recv.cc83 Mesh send_mesh; in LowerDTensorSendRecvsOps() local
H A Ddtensor_send_recv.cc228 const Mesh& send_mesh, mlir::TF::DTensorRecv dtensor_recv) { in LowerDTensorRecvFromCPUToTFOp()