Home
last modified time | relevance | path

Searched defs:original_layout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_allreduce_scatter_optimization.cc51 const Layout original_layout = all_scatter.input_layout(); in GetScatterGroupAssignment() local
87 const Layout original_layout = all_scatter.input_layout(); in ApplyOptimization() local
H A Dcollectives.cc107 const Layout& original_layout, const Layout& desired_layout, in EmitAllScatter()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dparams_test.cc494 std::vector<int64_t> original_layout( in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dcollective_lowering.cc631 const Layout original_layout = all_scatter.input_layout(); in LowerAllScatterOp() local