Searched defs:dtensor_send (Results 1 – 2 of 2) sorted by relevance
59 Status ValidateSendRecvLayoutConfiguration(mlir::TF::DTensorSend dtensor_send, in ValidateSendRecvLayoutConfiguration()268 auto dtensor_send = llvm::cast<mlir::TF::DTensorSend>(op); in ExpandOp() local395 auto dtensor_send = llvm::dyn_cast<mlir::TF::DTensorSend>(send_op); in ExpandOp() local
94 mlir::TF::DTensorSend dtensor_send, bool send_from_device_zero) { in LowerDTensorSendToXlaOp()198 mlir::TF::DTensorSend dtensor_send) { in LowerDTensorSendFromCPUToTFOp()