Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_hlo_test.cc45 void PrintTo(const ReduceLayout& reduce_layout, ::std::ostream* os) { in PrintTo()
81 const ReduceLayout& reduce_layout = GetParam(); in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_mixed_precision_reduce.cc84 const auto reduce_layout = ExtractRequiredSingleLayoutFromOp(reduce_op); in MaybeUpcastForReduction() local