Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_send_recv.cc93 const Layout& send_input_layout, mlir::Value send_input, in LowerDTensorSendToXlaOp()
197 const Layout& send_input_layout, mlir::Value send_input, in LowerDTensorSendFromCPUToTFOp()
H A Dlower_send_recv.cc100 const Layout send_input_layout = in LowerDTensorSendRecvsOps() local