| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 111 def collective_permute(x, source_target_pairs, name=None): argument
134 return gen_tpu_ops.collective_permute(x, source_target_pairs, name=name)
142 source_target_pairs = op.inputs[1][:, ::-1]
143 return [gen_tpu_ops.collective_permute(grad, source_target_pairs), None]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CollectivePermute.pbtxt | 12 name: "source_target_pairs"
34 source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` gets the outputs:
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 401 std::vector<std::pair<int64, int64>> source_target_pairs( in CreateFromProto() local
403 for (int i = 0; i < source_target_pairs.size(); i++) { in CreateFromProto()
404 source_target_pairs[i].first = proto.source_target_pairs(i).source(); in CreateFromProto()
405 source_target_pairs[i].second = proto.source_target_pairs(i).target(); in CreateFromProto()
408 CreateCollectivePermute(shape, operands(0), source_target_pairs); in CreateFromProto()
852 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CreateCollectivePermute() argument
854 shape, operand, source_target_pairs); in CreateCollectivePermute()
3417 HloInstruction::source_target_pairs() const { in source_target_pairs() function in xla::HloInstruction
3418 return Cast<HloCollectivePermuteInstruction>(this)->source_target_pairs(); in source_target_pairs()
|
| D | hlo_instructions.cc | 603 const std::vector<std::pair<int64, int64>>& source_target_pairs) in HloCollectivePermuteInstruction() argument
605 source_target_pairs_(source_target_pairs) { in HloCollectivePermuteInstruction()
611 for (const auto& pair : source_target_pairs()) { in ToProto()
624 for (const auto& pair : source_target_pairs()) { in ExtraAttributesToStringImpl()
637 return absl::c_equal(source_target_pairs(), in IdenticalSlowPath()
638 casted_other.source_target_pairs(), in IdenticalSlowPath()
648 shape, new_operands[0], source_target_pairs()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.h | 517 const std::vector<std::pair<int64, int64>>& source_target_pairs);
1525 const std::vector<std::pair<int64, int64>>& source_target_pairs() const;
|
| D | hlo_instructions.h | 374 const std::vector<std::pair<int64, int64>>& source_target_pairs);
376 const std::vector<std::pair<int64, int64>>& source_target_pairs() const { in source_target_pairs() function
|
| D | hlo.proto | 191 repeated SourceTarget source_target_pairs = 52; field
|
| D | hlo_parser_test.cc | 1424 ROOT root = f32[128,32]{0,1} collective-permute(input), source_target_pairs={{0,1},{1,2},{2,3}} in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 480 const std::vector<std::pair<int64, int64>>& source_target_pairs);
895 const std::vector<std::pair<int64, int64>>& source_target_pairs);
1607 const std::vector<std::pair<int64, int64>>& source_target_pairs);
|
| D | xla_builder.cc | 2216 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
2225 for (const auto& pair : source_target_pairs) { in CollectivePermute()
3253 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
3254 return operand.builder()->CollectivePermute(operand, source_target_pairs); in CollectivePermute()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 460 <b> `CollectivePermute(operand, source_target_pairs)` </b>
465 | `source_target_pairs` | `<int64, int64>` vector | A list of |
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 589 …argspec: "args=[\'input\', \'source_target_pairs\', \'name\'], varargs=None, keywords=None, defaul…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 589 …argspec: "args=[\'input\', \'source_target_pairs\', \'name\'], varargs=None, keywords=None, defaul…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 21014 func CollectivePermute(scope *Scope, input tf.Output, source_target_pairs tf.Output) (output tf.Out…
21021 input, source_target_pairs,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 5599 name: "source_target_pairs"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 13068 name: "source_target_pairs"
|
| D | ops_history.v2.pbtxt | 13010 name: "source_target_pairs"
|