Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dcollectives.cc52 const dtensor::Layout& src_layout, const dtensor::Layout& tgt_layout, in EmitAllGather()
200 const dtensor::Layout& tgt_layout, in EmitRelayout()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dcollective_lowering.cc444 const Layout tgt_layout = all_gather.output_layout(); in LowerAllGatherOp() local