Searched refs:CollectivePermute (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CollectivePermute.pbtxt | 2 graph_op_name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 198 HLO_MATCHER(CollectivePermute);
|
| D | hlo_parser_test.cc | 1419 "CollectivePermute", in CreateTestCases()
1420 R"(HloModule CollectivePermute in CreateTestCases()
1422 ENTRY CollectivePermute { in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 378 TEST_F(XlaBuilderTest, CollectivePermute) { in TEST_F() argument
381 CollectivePermute(x, {{0, 1}, {1, 2}, {2, 3}}); in TEST_F()
|
| D | xla_builder.h | 478 XlaOp CollectivePermute(
893 friend XlaOp CollectivePermute(
1605 XlaOp CollectivePermute(
|
| D | xla_builder.cc | 2214 XlaOp XlaBuilder::CollectivePermute( in CollectivePermute() function in xla::XlaBuilder
3251 XlaOp CollectivePermute( in CollectivePermute() function
3254 return operand.builder()->CollectivePermute(operand, source_target_pairs); in CollectivePermute()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 452 ## CollectivePermute section in Operation Semantics
455 [`XlaBuilder::CollectivePermute`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xl…
457 CollectivePermute is a collective operation that sends and receives data cross
460 <b> `CollectivePermute(operand, source_target_pairs)` </b>
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 588 name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 588 name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 21014 func CollectivePermute(scope *Scope, input tf.Output, source_target_pairs tf.Output) (output tf.Out… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 5593 name: "CollectivePermute"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 13062 name: "CollectivePermute"
|
| D | ops_history.v2.pbtxt | 13004 name: "CollectivePermute"
|