Home
last modified time | relevance | path

Searched defs:reduce_scatter (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dall_reduce_blueconnect_test.cc73 auto reduce_scatter = AllOf(op::Shape("f32[4]"), op::ReduceScatter(bitcast), in TEST_F() local
159 auto reduce_scatter = AllOf(op::Shape("(f32[4], f32[8])"), in TEST_F() local
Dall_reduce_blueconnect.cc225 HloInstruction* reduce_scatter = in TryDecomposeAllReduce() local
/external/tensorflow/tensorflow/dtensor/mlir/
Ddtensor_allreduce_scatter_optimization.cc138 auto reduce_scatter = builder.create<mlir::TF::DTensorReduceScatterOp>( in ApplyOptimization() local
/external/tensorflow/tensorflow/dtensor/mlir/utils/
Dcollective_lowering.cc325 mlir::TF::DTensorReduceScatterOp reduce_scatter) { in LowerReduceScatterOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc1186 auto reduce_scatter = Cast<HloReduceScatterInstruction>(instruction); in ImportInstructionImpl() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6111 auto reduce_scatter = rewriter.create<ReduceScatterOp>( in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc10018 auto reduce_scatter = AllOf(op::Shape("f32[16,32,24,512]"), in TEST_F() local