Home
last modified time | relevance | path

Searched defs:update_layout (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreduce_scatter_decomposer.h33 : update_layout_(update_layout) {} in update_layout_() argument
H A Dcollective_decomposer_utils.cc38 std::function<void(Shape &)> update_layout) { in CreateStartIndicesForCollectiveDecomposition()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executable_interface.cc63 auto update_layout = [this](xla::Shape* subshape, in AllocateOutputMemoryWithInputReuse() local