Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dall_reduce_contiguous.cc56 HloInstruction* new_all_reduce = in ReplaceWithContiguousAllReduce() local
H A Dwhile_loop_all_reduce_code_motion.cc635 HloInstruction* new_all_reduce = in CreateSinkedAllReduces() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dall_reduce_blueconnect.cc232 HloInstruction* new_all_reduce = in TryDecomposeAllReduce() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_allreduce_sum_optimization.cc387 auto new_all_reduce = builder.create<mlir::TF::DTensorAllReduceOp>( in ExtractAllReduceFromWhileOp() local