Searched defs:copy_to_mesh (Results 1 – 4 of 4) sorted by relevance
86 auto copy_to_mesh = in CloneOpToCluster() local178 mlir::LogicalResult LowerToSendRecv(mlir::TF::CopyToMeshOp copy_to_mesh, in LowerToSendRecv()299 mlir::TF::CopyToMeshOp copy_to_mesh = in ReplaceCopyToMeshWithVirtualSendRecv() local
59 } else if (auto copy_to_mesh = llvm::dyn_cast<mlir::TF::CopyToMeshOp>(op)) { in WrapDeviceCluster() local
130 def copy_to_mesh( function
182 def copy_to_mesh(self, tensor, new_layout, source_layout=None) -> ops.Tensor: member in DTensorDevice