Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsoftmax_spmd_expander.cc49 mlir::Operation* local_reduce; in ComputeGlobalReduce() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_test.cc3443 auto local_reduce = in TEST_F() local
3498 auto local_reduce = in TEST_F() local
H A Dspmd_partitioner.cc3522 auto local_reduce = b_.AddInstruction(HloInstruction::CreateReduce( in HandleReduce() local