Searched refs:send_input_layout (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | dtensor_send_recv.cc | 93 const Layout& send_input_layout, mlir::Value send_input, in LowerDTensorSendToXlaOp() argument
95 const bool send_from_cpu = !send_input_layout.mesh().is_tpu_mesh(); in LowerDTensorSendToXlaOp()
124 GetDeviceOrdinal(send_input_layout.mesh(), loc, send_func, &builder)); in LowerDTensorSendToXlaOp()
197 const Layout& send_input_layout, mlir::Value send_input, in LowerDTensorSendFromCPUToTFOp() argument
207 send_input_layout.mesh().local_devices(); in LowerDTensorSendFromCPUToTFOp()
|
| D | dtensor_send_recv.h | 125 const Layout& send_input_layout, mlir::Value send_input,
130 const Layout& send_input_layout, mlir::Value send_input,
|
| D | lower_send_recv.cc | 100 const Layout send_input_layout = in LowerDTensorSendRecvsOps() local
113 LowerDTensorSendToXlaOp(send_input_layout, send_op.input(), send_op, in LowerDTensorSendRecvsOps()
|