| /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()
89 AllOf(op::Reshape(op::AllGather(_)), op::Shape("s32[2,8,1,1]"))); in TEST_F()
109 AllOf(op::Reshape(op::AllGather(_)), op::Shape("s32[2,8,1,1]"))); in TEST_F()
128 EXPECT_THAT(reshape, AllOf(op::AllGather(op::Reshape(op::Reshape(_))), in TEST_F()
148 EXPECT_THAT(reshape, AllOf(op::AllGather(op::Reshape(op::Reshape(_))), in TEST_F()
169 AllOf(op::Reshape(op::AllGather(_)), op::Shape("s32[2,8,1,1]"))); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser_test.cc | 1597 "AllGather", in CreateTestCases()
1598 R"(HloModule AllGather in CreateTestCases()
1600 ENTRY AllGather { in CreateTestCases()
1610 R"(HloModule AllGather in CreateTestCases()
1612 ENTRY AllGather { in CreateTestCases()
|
| D | hlo_matchers.h | 206 HLO_MATCHER(AllGather);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 541 // Common base class for AllReduce, AllGather, and AllToAll.
554 // AllGather is cross replica if channel_id is not set.
|
| D | hlo_ops_base.td | 660 string summary = "AllGather operator";
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 59 ops.def("AllGather", &AllGather, py::arg("operand"), in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 388 AllGather(x, /*all_gather_dimension=*/0, /*shard_count=*/4); in TEST_F()
399 AllGather(x, /*all_gather_dimension=*/1, /*shard_count=*/4); in TEST_F()
|
| D | xla_builder.h | 733 XlaOp AllGather(
1297 friend XlaOp AllGather(XlaOp operand, int64 all_gather_dimension,
2188 XlaOp AllGather(
|
| D | xla_builder.cc | 2714 XlaOp XlaBuilder::AllGather(XlaOp operand, int64 all_gather_dimension, in AllGather() function in xla::XlaBuilder
4653 XlaOp AllGather(const XlaOp operand, int64 all_gather_dimension, in AllGather() function
4659 return operand.builder()->AllGather(operand, all_gather_dimension, in AllGather()
|
| /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/libabigail/tests/data/test-annotate/ |
| D | test20-pr19025-libvtkParallelCore-6.1.so.abi | 174 <!-- vtkCommunicator::AllGather(vtkDataArray*, vtkDataArray*) -->
2169 <!-- int vtkCommunicator::AllGather(vtkDataArray*, vtkDataArray*) -->
2170 …<function-decl name='AllGather' mangled-name='_ZN15vtkCommunicator9AllGatherEP12vtkDataArrayS1_' f…
|
| /external/libabigail/tests/data/test-read-dwarf/ |
| D | test20-pr19025-libvtkParallelCore-6.1.so.abi | 1337 …<function-decl name='AllGather' mangled-name='_ZN15vtkCommunicator9AllGatherEP12vtkDataArrayS1_' f…
|