Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/mlir/
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
Dcollectives.cc107 const Layout& original_layout, const Layout& desired_layout, in EmitAllScatter()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dparams_test.cc494 std::vector<int64_t> original_layout( in XLA_TEST_F() local
/external/tensorflow/tensorflow/dtensor/mlir/utils/
Dcollective_lowering.cc631 const Layout original_layout = all_scatter.input_layout(); in LowerAllScatterOp() local