Home
last modified time | relevance | path

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

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