Home
last modified time | relevance | path

Searched refs:source_target_pairs (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py113 def collective_permute(x, source_target_pairs, name=None): argument
136 return gen_tpu_ops.collective_permute(x, source_target_pairs, name=name)
144 source_target_pairs = op.inputs[1][:, ::-1]
145 return [gen_tpu_ops.collective_permute(grad, source_target_pairs), None]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_CollectivePermute.pbtxt12 name: "source_target_pairs"
34 source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` gets the outputs:
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DCollectivePermute.pbtxt8 name: "source_target_pairs"
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc244 for (const auto& src_dest : instr->source_target_pairs()) { in ExecuteOnStream()
259 if (absl::c_none_of(instr->source_target_pairs(), in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.h127 source_target_pairs);
Dhlo_function_importer.cc265 ConvertSourceTargetPairs(instruction->source_target_pairs())); in ImportInstruction()
809 source_target_pairs) { in ConvertSourceTargetPairs() argument
810 std::vector<int64_t> attr(source_target_pairs.size() * 2); in ConvertSourceTargetPairs()
811 for (auto p : llvm::enumerate(source_target_pairs)) { in ConvertSourceTargetPairs()
Dmlir_hlo_to_hlo.cc156 llvm::Optional<mlir::DenseIntElementsAttr> source_target_pairs) { in Convert_source_target_pairs() argument
157 return Convert_Nx2_attribute(source_target_pairs); in Convert_source_target_pairs()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dops.mlir170 source_target_pairs = dense<[[0, 1], [0, 2], [2, 3]]> : tensor<3x2xi64>
180 source_target_pairs = dense<[[0, 1], [1, 2], [2, 1]]> : tensor<3x2xi64>
188 // expected-error@+1 {{expect source_target_pairs attribute to be of rank 2, but got rank 1}}
190 source_target_pairs = dense<[0, 1]> : tensor<2xi64>
198 // expected-error@+1 {{expect source_target_pairs attribute of shape (N, 2), but got (2, 3)}}
200 source_target_pairs = dense<[[0, 1, 2], [3, 4, 5]]> : tensor<2x3xi64>
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc430 std::vector<std::pair<int64, int64>> source_target_pairs( in CreateFromProto() local
436 for (int i = 0; i < source_target_pairs.size(); i++) { in CreateFromProto()
437 source_target_pairs[i].first = proto.source_target_pairs(i).source(); in CreateFromProto()
438 source_target_pairs[i].second = proto.source_target_pairs(i).target(); in CreateFromProto()
441 source_target_pairs, channel_id); in CreateFromProto()
932 const std::vector<std::pair<int64, int64>>& source_target_pairs, in CreateCollectivePermute() argument
935 shape, operand, source_target_pairs, channel_id); in CreateCollectivePermute()
3762 HloInstruction::source_target_pairs() const { in source_target_pairs() function in xla::HloInstruction
3763 return Cast<HloCollectivePermuteInstruction>(this)->source_target_pairs(); in source_target_pairs()
Dhlo_instructions.cc654 const std::vector<std::pair<int64, int64>>& source_target_pairs, in HloCollectivePermuteInstruction() argument
657 source_target_pairs_(source_target_pairs) { in HloCollectivePermuteInstruction()
663 for (const auto& pair : source_target_pairs()) { in ToProto()
677 for (const auto& pair : source_target_pairs()) { in ExtraAttributesToStringImpl()
691 absl::c_equal(source_target_pairs(), in IdenticalSlowPath()
692 casted_other.source_target_pairs(), in IdenticalSlowPath()
702 shape, new_operands[0], source_target_pairs(), channel_id()); in CloneWithNewOperandsImpl()
Dhlo_instruction.h654 const std::vector<std::pair<int64, int64>>& source_target_pairs,
1708 const std::vector<std::pair<int64, int64>>& source_target_pairs() const;
Dhlo_instructions.h420 const std::vector<std::pair<int64, int64>>& source_target_pairs,
423 const std::vector<std::pair<int64, int64>>& source_target_pairs() const { in source_target_pairs() function
Dhlo.proto196 repeated SourceTarget source_target_pairs = 52; field
Dhlo_verifier.cc282 for (const auto& p : hlo->source_target_pairs()) { in HandleCollectivePermute()
Dhlo_parser_test.cc1502 ROOT root = f32[128,32]{0,1} collective-permute(input), source_target_pairs={{0,1},{1,2},{2,3}}
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dhlo_ops.cc227 auto type = op.source_target_pairs().getType().dyn_cast<RankedTensorType>(); in Verify()
239 for (auto i = op.source_target_pairs().begin(), in Verify()
240 e = op.source_target_pairs().end(); in Verify()
Dhlo_ops.td808 I64ElementsAttr:$source_target_pairs
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h526 const std::vector<std::pair<int64, int64>>& source_target_pairs);
933 const std::vector<std::pair<int64, int64>>& source_target_pairs);
1695 const std::vector<std::pair<int64, int64>>& source_target_pairs);
Dxla_builder.cc2246 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
2255 for (const auto& pair : source_target_pairs) { in CollectivePermute()
3323 const std::vector<std::pair<int64, int64>>& source_target_pairs) { in CollectivePermute() argument
3324 return operand.builder()->CollectivePermute(operand, source_target_pairs); in CollectivePermute()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir224 source_target_pairs = dense<[[0, 1], [1, 2], [2, 3]]> : tensor<3x2xi64>
230 …ESULT:%.*]] = f32[128,32] collective-permute(f32[128,32] [[ARG]]), source_target_pairs={{\{\{}}0,1…
Dimport.hlotxt121 …// CHECK-NEXT: "xla_hlo.collective_permute"([[ARG]]) {name = {{.*}}, source_target_pairs = dense<…
122 ROOT root = f32[128,32]{0,1} collective-permute(%input), source_target_pairs={{0,1},{1,2},{2,3}}
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md534 <b> `CollectivePermute(operand, source_target_pairs)` </b>
539 | `source_target_pairs` | `<int64, int64>` vector | A list of |
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt709 …argspec: "args=[\'input\', \'source_target_pairs\', \'name\'], varargs=None, keywords=None, defaul…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt709 …argspec: "args=[\'input\', \'source_target_pairs\', \'name\'], varargs=None, keywords=None, defaul…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go38624 func CollectivePermute(scope *Scope, input tf.Output, source_target_pairs tf.Output) (output tf.Out…
38631 input, source_target_pairs,

12