Searched defs:reduce_scatter (Results 1 – 7 of 7) sorted by relevance
73 auto reduce_scatter = AllOf(op::Shape("f32[4]"), op::ReduceScatter(bitcast), in TEST_F() local159 auto reduce_scatter = AllOf(op::Shape("(f32[4], f32[8])"), in TEST_F() local
225 HloInstruction* reduce_scatter = in TryDecomposeAllReduce() local
138 auto reduce_scatter = builder.create<mlir::TF::DTensorReduceScatterOp>( in ApplyOptimization() local
325 mlir::TF::DTensorReduceScatterOp reduce_scatter) { in LowerReduceScatterOp()
1186 auto reduce_scatter = Cast<HloReduceScatterInstruction>(instruction); in ImportInstructionImpl() local
6111 auto reduce_scatter = rewriter.create<ReduceScatterOp>( in matchAndRewrite() local
10018 auto reduce_scatter = AllOf(op::Shape("f32[16,32,24,512]"), in TEST_F() local