| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_all_reduce_code_motion_test.cc | 102 Each(Not(op::AllReduce()))); in TEST_F()
110 return Value(instruction, op::AllReduce()); in TEST_F()
201 Each(Not(op::AllReduce()))); in TEST_F()
206 return Value(instruction, op::AllReduce()); in TEST_F()
280 Each(Not(op::AllReduce()))); in TEST_F()
396 Each(Not(op::AllReduce()))); in TEST_F()
404 return Value(instruction, op::AllReduce()); in TEST_F()
483 Matches(op::AllReduce())), in TEST_F()
494 Each(Not(op::AllReduce()))); in TEST_F()
567 Each(Not(op::AllReduce()))); in TEST_F()
[all …]
|
| D | ar_crs_combiner_test.cc | 461 op::Tuple(op::AllReduce(op::Convert(op::Parameter())), in TEST_F()
462 op::AllReduce(op::Convert(op::Constant())))); in TEST_F()
512 op::Tuple(op::AllReduce(op::Convert(op::Parameter())))); in TEST_F()
581 op::Tuple(op::AllReduce(op::Bitcast(op::Parameter())), in TEST_F()
582 op::AllReduce(op::Bitcast(op::Parameter())))); in TEST_F()
651 op::Tuple(op::AllReduce(op::Multiply(op::Parameter(), op::Constant())), in TEST_F()
652 op::AllReduce(op::Multiply(op::Parameter(), op::Constant())))); in TEST_F()
694 op::Tuple(op::AllReduce(op::Multiply(op::Parameter(), op::Constant())))); in TEST_F()
777 op::AllReduce(op::Add(op::Divide(op::Constant(), op::Constant()), in TEST_F()
779 op::AllReduce(op::Add(op::Divide(op::Constant(), op::Constant()), in TEST_F()
[all …]
|
| D | all_gather_decomposer_test.cc | 56 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
79 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
102 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
127 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
154 op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
|
| D | all_reduce_simplifier_test.cc | 90 m::Parameter(0), m::AllReduce(m::Parameter(1)), in TEST_F()
123 m::AllReduce(m::Parameter(0)), in TEST_F()
167 m::Parameter(0), m::AllReduce(m::Parameter(1))))); in TEST_F()
|
| D | all_reduce_combiner_test.cc | 421 AllOf(op::AllReduce(op::Parameter(0), op::Parameter(0)), in TEST_F()
425 AllOf(op::AllReduce(op::Parameter(1), op::Parameter(1)), in TEST_F()
469 auto crs0 = op::AllReduce(op::Parameter(0), op::AllReduce(op::Parameter(1))); in TEST_F()
470 auto add = op::Add(op::AllReduce(op::GetTupleElement(crs0, 0)), in TEST_F()
472 auto crs1 = op::AllReduce(add, op::GetTupleElement(crs0)); in TEST_F()
|
| D | conditional_code_motion_test.cc | 621 op::Convert(op::AllReduce(op::GetTupleElement(op::Conditional()))), in TEST_F()
623 op::AllReduce(op::GetTupleElement(op::Conditional()))))))); in TEST_F()
705 EXPECT_THAT(root, AllOf(op::Tuple(op::Convert(op::AllReduce( in TEST_F()
1141 op::Convert(op::AllReduce(op::GetTupleElement(op::Conditional()))), in TEST_F()
1143 op::AllReduce(op::GetTupleElement(op::Conditional()))))))); in TEST_F()
|
| D | hlo_computation_test.cc | 701 ElementsAre(op::Parameter(), op::AllReduce(), op::AllReduce(), in TEST_F()
|
| D | hlo_matchers.h | 207 HLO_MATCHER(AllReduce);
|
| D | hlo.proto | 130 // optionally for collective instructions (AllReduce, CollectivePermute,
|
| D | pattern_matcher.h | 2013 XLA_UNOP_PATTERN(AllReduce)
|
| D | hlo_parser_test.cc | 1523 "AllReduce", in CreateTestCases()
|
| D | sharding_propagation_test.cc | 1566 TEST_P(ParameterizedMetadataTest, AllReduce) { in TEST_P() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 100 EXPECT_THAT(root, AllOf(op::Copy(op::AllReduce( in TEST_F()
119 EXPECT_THAT(root, op::Copy(AllOf(op::Copy(op::AllReduce(op::Select( in TEST_F()
143 op::AllReduce(op::Select( in TEST_F()
165 op::Copy(op::AllReduce(AllOf( in TEST_F()
187 op::Copy(op::Copy(op::AllReduce(AllOf( in TEST_F()
255 op::Copy(op::AllReduce(op::Select( in TEST_F()
259 op::Copy(op::AllReduce(op::Select( in TEST_F()
311 op::Copy(op::AllReduce(op::DynamicUpdateSlice( in TEST_F()
445 op::AllReduce(op::Reduce( in TEST_F()
507 root, AllOf(op::Shape("f32[2,3]{1,0}"), op::AllReduce(op::Parameter(0)))); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_CollectiveReduce.pbtxt | 4 name: "collective.AllReduce"
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_structs.td | 65 // optionally for collective instructions (AllReduce, CollectivePermute,
|
| D | lhlo_ops.td | 541 // Common base class for AllReduce, AllGather, and AllToAll.
|
| D | hlo_ops_base.td | 670 string summary = "AllReduce operator";
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | steps_db.proto | 71 // all_reduce_id, they will be 'Allreduce'd'. If empty, AllReduce will not be
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | simple_orc_jit.cc | 254 REGISTER_CPU_RUNTIME_SYMBOL(AllReduce); in RegisterKnownJITSymbols()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 70 &AllReduce), in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 740 XlaOp AllReduce(
1303 friend XlaOp AllReduce(XlaOp operand, const XlaComputation& computation,
2214 XlaOp AllReduce(XlaOp operand, const XlaComputation& computation,
|
| D | xla_builder.cc | 2776 return AllReduce(operand, computation, replica_groups, in CrossReplicaSum()
2781 XlaOp XlaBuilder::AllReduce(XlaOp operand, const XlaComputation& computation, in AllReduce() function in xla::XlaBuilder
4669 XlaOp AllReduce(const XlaOp operand, const XlaComputation& computation, in AllReduce() function
4673 return operand.builder()->AllReduce(operand, computation, replica_groups, in AllReduce()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 584 value_map[op] = xla::AllReduce(operand, computation, replica_groups, in ExportXlaOp()
590 xla::AllReduce(operand, computation, replica_groups, channel_id); in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 199 HloModule AllReduce
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 71 ## AllReduce section in Operation Semantics
74 [`XlaBuilder::AllReduce`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builde…
78 <b> `AllReduce(operand, computation, replica_group_ids, channel_id)` </b>
108 Computing the result of `AllReduce` requires having one input from each replica,
109 so if one replica executes a `AllReduce` node more times than another, then the
901 Performs `AllReduce` with a summation computation.
|