Searched refs:all_reduce_done (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule_test.cc | 211 HloInstruction* all_reduce_done = in TEST_F() local
217 f32_2x2_, HloOpcode::kAdd, add3, all_reduce_done)); in TEST_F()
234 EXPECT_TRUE(order.ExecutesBefore(add3, all_reduce_done)); in TEST_F()
235 EXPECT_TRUE(order.ExecutesBefore(all_reduce_done, add4)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis.h | 284 bool UpdateAllReduceDoneValueSet(HloInstruction* all_reduce_done);
|
| D | hlo_dataflow_analysis.cc | 1052 HloInstruction* all_reduce_done) { in UpdateAllReduceDoneValueSet() argument
1053 CHECK_EQ(all_reduce_done->opcode(), HloOpcode::kAllReduceDone); in UpdateAllReduceDoneValueSet()
1056 for (auto& pair : GetInstructionValueSet(all_reduce_done)) { in UpdateAllReduceDoneValueSet()
1066 GetValueSet(all_reduce_done->operand(0), operand_index); in UpdateAllReduceDoneValueSet()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo_gpu/IR/ |
| D | lhlo_gpu_ops.td | 187 LHLOGPU_Op<"all_reduce_done", [SameVariadicOperandSize]> {
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 250 // CHECK: "lmhlo_gpu.all_reduce_done"([[TOKEN]], [[OUTPUT]])
284 // CHECK: "lmhlo_gpu.all_reduce_done"([[TOKEN]], [[OUTPUT0]], [[OUTPUT1]])
|