| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | canonicalize_all_gather_for_cse_test.cc | 69 AllOf(op::Reshape(op::AllGather(_)), op::Shape("s32[2,8]"))); in TEST_F()
88 EXPECT_THAT(reshape, op::Reshape(op::AllGather(op::Parameter()))); in TEST_F()
107 EXPECT_THAT(reshape, op::Reshape(op::AllGather(op::Parameter()))); in TEST_F()
126 EXPECT_THAT(reshape, op::Reshape(op::AllGather(op::Parameter()))); in TEST_F()
145 EXPECT_THAT(reshape, AllOf(op::AllGather(op::Reshape(op::Reshape(_))), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | all_reduce_blueconnect_test.cc | 77 auto all_gather = AllOf(op::Shape("f32[16]"), op::AllGather(all_reduce), in TEST_F()
119 auto all_gather0 = AllOf(op::Shape("f32[4]"), op::AllGather(all_reduce), in TEST_F()
122 auto all_gather1 = AllOf(op::Shape("f32[16]"), op::AllGather(bitcast2), in TEST_F()
167 op::AllGather(op::GetTupleElement(all_reduce, 0), in TEST_F()
|
| D | jitrt_custom_calls.cc | 1858 struct AllGather { struct
1866 static AllGather Handler() { return AllGather(); } in Handler() argument
1870 Error AllGather::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
1902 static bool AllGather(runtime::KernelContext* ctx, void** args, void** attrs) { in AllGather() function
1913 .To<RuntimeChecks()>(AllGather::Handler()) in AllGather()
2199 lib.Insert("xla.gpu.all_gather", &xla::gpu::AllGather); in JitRtGpuCustomCalls()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_gather_broadcast_reorder_test.cc | 42 m::Broadcast(m::AllGather(m::Parameter()))); in RunPass()
46 m::Reshape(m::Broadcast(m::AllGather(m::Reshape(m::Parameter()))))); in RunPass()
|
| D | all_gather_combiner_test.cc | 81 op::AllGather(op::Parameter(0), op::Parameter(1)); in TEST_F()
117 op::AllGather(op::Parameter(0), op::Parameter(1), op::Parameter(4)); in TEST_F()
119 op::AllGather(op::Parameter(2), op::Parameter(3)); in TEST_F()
|
| D | hlo_parser_test.cc | 1799 "AllGather", in CreateTestCases()
1800 R"(HloModule AllGather, entry_computation_layout={(f32[128,32]{0,1})->f32[128,128]{0,1}} in CreateTestCases()
1802 ENTRY AllGather { in CreateTestCases()
1812 R"(HloModule AllGather, entry_computation_layout={(f32[128,32]{0,1})->f32[128,128]{0,1}} in CreateTestCases()
1814 ENTRY AllGather { in CreateTestCases()
|
| D | hlo_matchers.h | 243 HLO_MATCHER(AllGather);
|
| D | hlo_rematerialization_test.cc | 1116 op::Bitcast(op::AllGather(op::Broadcast(_))), _))); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_control_flow_flattening_test.cc | 372 TEST_F(HloControlFlowFlatteningTest, AllGather) { in TEST_F() argument
624 metadata={op_type="AllGather" op_name="ag0"} in TEST_F()
627 metadata={op_type="AllGather" op_name="ag0"} in TEST_F()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | Passes.td | 175 let summary = "Converts logical AllGather ops into physical AllGather ops.";
|
| /external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
| D | ops.pyi | 64 def AllGather(
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 73 ops.def("AllGather", &AllGather, py::arg("operand"), in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 390 AllGather(x, /*all_gather_dimension=*/0, /*shard_count=*/4); in TEST_F()
401 AllGather(x, /*all_gather_dimension=*/1, /*shard_count=*/4); in TEST_F()
|
| D | xla_builder.h | 747 XlaOp AllGather(
1363 friend XlaOp AllGather(XlaOp operand, int64_t all_gather_dimension,
2325 XlaOp AllGather(XlaOp operand, int64_t all_gather_dimension,
|
| D | xla_builder.cc | 2868 XlaOp XlaBuilder::AllGather(XlaOp operand, int64_t all_gather_dimension, in AllGather() function in xla::XlaBuilder
4653 XlaOp AllGather(const XlaOp operand, int64_t all_gather_dimension, in AllGather() function
4659 return operand.builder()->AllGather(operand, all_gather_dimension, in AllGather()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo/IR/ |
| D | lhlo_ops.td | 1086 // Common base class for AllReduce, AllGather, and AllToAll.
1099 // AllGather is cross replica if channel_id is not set.
1105 let summary = "AllGather operator";
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 372 // ReduceScatter-AllReduce-AllGather sequence, with the initial ReduceScatter
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 32 ## AllGather section in Operation Semantics
35 [`XlaBuilder::AllGather`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builde…
39 <b> `AllGather(operand, all_gather_dim, shard_count, replica_group_ids,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 758 value_map[op] = xla::AllGather(operand, all_gather_dim, shard_count, in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | StablehloOps.td | 951 string summary = "AllGather operator";
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 1017 string summary = "AllGather operator";
|
| /external/libabigail/tests/data/test-annotate/ |
| D | test20-pr19025-libvtkParallelCore-6.1.so.abi | 174 <!-- vtkCommunicator::AllGather(vtkDataArray*, vtkDataArray*) -->
6244 <!-- int vtkCommunicator::AllGather(const vtkIdType*, vtkIdType*, vtkIdType) -->
6245 …<function-decl name='AllGather' mangled-name='_ZN15vtkCommunicator9AllGatherEPKxPxx' filepath='/tm…
6516 <!-- int vtkCommunicator::AllGather(vtkDataArray*, vtkDataArray*) -->
6517 …<function-decl name='AllGather' mangled-name='_ZN15vtkCommunicator9AllGatherEP12vtkDataArrayS1_' f…
6834 <!-- int vtkCommunicator::AllGather(const int*, int*, vtkIdType) -->
6835 …<function-decl name='AllGather' mangled-name='_ZN15vtkCommunicator9AllGatherEPKiPix' filepath='/tm…
9138 <!-- int vtkMultiProcessController::AllGather(const int*, int*, vtkIdType) -->
9139 …<function-decl name='AllGather' mangled-name='_ZN25vtkMultiProcessController9AllGatherEPKiPix' fil…
|
| /external/libabigail/tests/data/test-read-dwarf/ |
| D | test20-pr19025-libvtkParallelCore-6.1.so.abi | 3870 …<function-decl name='AllGather' mangled-name='_ZN15vtkCommunicator9AllGatherEPKxPxx' filepath='/tm…
4033 …<function-decl name='AllGather' mangled-name='_ZN15vtkCommunicator9AllGatherEP12vtkDataArrayS1_' f…
4225 …<function-decl name='AllGather' mangled-name='_ZN15vtkCommunicator9AllGatherEPKiPix' filepath='/tm…
5610 …<function-decl name='AllGather' mangled-name='_ZN25vtkMultiProcessController9AllGatherEPKiPix' fil…
|