| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 626 auto left_halo = AllOf(op::CollectivePermute(op::Slice(lhs)), in TEST_F()
628 auto right_halo = AllOf(op::CollectivePermute(op::Slice(lhs)), in TEST_F()
865 op::CollectivePermute(op::Slice(sharded_input))); in TEST_F()
901 op::CollectivePermute(op::Slice(op::Parameter(0)))); in TEST_F()
903 AllOf(op::Shape("f32[2,2]"), op::CollectivePermute(op::Parameter(0))); in TEST_F()
944 op::CollectivePermute(op::Slice(sharded_input))); in TEST_F()
986 op::CollectivePermute(op::Slice(sharded_input))); in TEST_F()
988 op::CollectivePermute(op::Slice(sharded_input))); in TEST_F()
1035 op::CollectivePermute(op::Slice(sharded_input))); in TEST_F()
1037 op::CollectivePermute(op::Slice(sharded_input))); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectivePermute.pbtxt | 2 name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CollectivePermute.pbtxt | 2 name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CollectivePermute.pbtxt | 2 graph_op_name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | cross_replica_ops.cc | 120 xla::CollectivePermute(ctx->Input(0), source_target_pairs)); in Compile()
|
| /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 | hlo_ops_base.td | 955 string summary = "CollectivePermute operator";
958 CollectivePermute is a collective operation that sends and receives data
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | simple_orc_jit.cc | 255 REGISTER_CPU_RUNTIME_SYMBOL(CollectivePermute); in RegisterKnownJITSymbols()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 216 HLO_MATCHER(CollectivePermute);
|
| D | hlo_parser_test.cc | 1660 "CollectivePermute", in CreateTestCases()
1661 R"(HloModule CollectivePermute in CreateTestCases()
1663 ENTRY CollectivePermute { in CreateTestCases()
|
| D | hlo.proto | 130 // optionally for collective instructions (AllReduce, CollectivePermute,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 79 ops.def("CollectivePermute", &CollectivePermute, py::arg("operand"), in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 438 TEST_F(XlaBuilderTest, CollectivePermute) { in TEST_F() argument
441 CollectivePermute(x, {{0, 1}, {1, 2}, {2, 3}}); in TEST_F()
|
| D | xla_builder.h | 756 XlaOp CollectivePermute(
1315 friend XlaOp CollectivePermute(
2241 XlaOp CollectivePermute(
|
| D | xla_builder.cc | 2996 XlaOp XlaBuilder::CollectivePermute( in CollectivePermute() function in xla::XlaBuilder
4694 XlaOp CollectivePermute( in CollectivePermute() function
4697 return operand.builder()->CollectivePermute(operand, source_target_pairs); in CollectivePermute()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 278 // CollectivePermute op patterns.
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 569 ## CollectivePermute section in Operation Semantics
572 [`XlaBuilder::CollectivePermute`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xl…
574 CollectivePermute is a collective operation that sends and receives data cross
577 <b> `CollectivePermute(operand, source_target_pairs)` </b>
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 218 def TF_CollectivePermuteOp : TF_Op<"CollectivePermute", []> {
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 776 name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 776 name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 3858 %0 = "tf.CollectivePermute"(%arg0, %source_target_pairs) {
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 39408 func CollectivePermute(scope *Scope, input tf.Output, source_target_pairs tf.Output) (output tf.Out… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7739 name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 13004 name: "CollectivePermute"
|