Home
last modified time | relevance | path

Searched refs:indices_are_sorted (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dhlo-legalize-gather-to-torch-index-select.mlir10 …e<2> : tensor<1xi64>, start_index_map = dense<0> : tensor<1xi64>}, indices_are_sorted = false, sli…
23 … 1]> : tensor<2xi64>, start_index_map = dense<0> : tensor<1xi64>}, indices_are_sorted = false, sli…
32 …e<2> : tensor<1xi64>, start_index_map = dense<0> : tensor<1xi64>}, indices_are_sorted = false, sli…
39 …e<2> : tensor<1xi64>, start_index_map = dense<0> : tensor<1xi64>}, indices_are_sorted = false, sli…
Dcanonicalize.mlir1023 indices_are_sorted = false,
1038 indices_are_sorted = false,
1053 indices_are_sorted = false,
1252 }) {indices_are_sorted = false,
1275 }) {indices_are_sorted = false,
1299 }) {indices_are_sorted = false,
1323 }) {indices_are_sorted = false,
1347 }) {indices_are_sorted = false,
1370 }) {indices_are_sorted = false,
1395 }) {indices_are_sorted = false,
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py483 indices_are_sorted=False, name=None): argument
489 indices_are_sorted=indices_are_sorted,
494 dimension_numbers, indices_are_sorted=False, name=None): argument
501 indices_are_sorted=indices_are_sorted,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dgpu_ops.mlir13 // CHECK: indices_are_sorted = false
29 indices_are_sorted = false,
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc201 gather->indices_are_sorted())); in ParititonPassthroughOperand()
250 gather->gather_slice_sizes(), gather->indices_are_sorted())); in ParititonTrivialIndexedOperandDimension()
431 gather->gather_slice_sizes(), gather->indices_are_sorted())); in PartitionIndexParallelDimensions()
625 scatter->to_apply(), dnums, scatter->indices_are_sorted(), in HandleScatter()
658 scatter->indices_are_sorted(), scatter->unique_indices())); in HandleScatter()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h182 absl::Span<const int64> slice_sizes, bool indices_are_sorted) override;
187 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted,
Dmlir_hlo_builder.cc290 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in GatherInternal() argument
303 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal() argument
310 builder_.getBoolAttr(indices_are_sorted), in ScatterInternal()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc2726 absl::Span<const int64> slice_sizes, bool indices_are_sorted) in HloGatherInstruction() argument
2728 indices_are_sorted_(indices_are_sorted) { in HloGatherInstruction()
2780 proto.set_indices_are_sorted(indices_are_sorted()); in ToProto()
2789 if (indices_are_sorted()) { in ExtraAttributesToStringImpl()
2804 indices_are_sorted() == casted_other.indices_are_sorted(); in IdenticalSlowPath()
2813 gather_slice_sizes(), indices_are_sorted()); in CloneWithNewOperandsImpl()
2820 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in HloScatterInstruction() argument
2823 indices_are_sorted_(indices_are_sorted), in HloScatterInstruction()
2878 proto.set_indices_are_sorted(indices_are_sorted()); in ToProto()
2887 if (indices_are_sorted()) { in ExtraAttributesToStringImpl()
[all …]
Dhlo_instructions.h1658 absl::Span<const int64> slice_sizes, bool indices_are_sorted);
1666 bool indices_are_sorted() const { return indices_are_sorted_; } in indices_are_sorted() function
1667 void set_indices_are_sorted(bool indices_are_sorted) { in set_indices_are_sorted() argument
1668 indices_are_sorted_ = indices_are_sorted; in set_indices_are_sorted()
1705 bool indices_are_sorted, bool unique_indices);
1710 bool indices_are_sorted() const { return indices_are_sorted_; } in indices_are_sorted() function
1711 void set_indices_are_sorted(bool indices_are_sorted) { in set_indices_are_sorted() argument
1712 indices_are_sorted_ = indices_are_sorted; in set_indices_are_sorted()
Dhlo_instruction.cc676 proto.indices_are_sorted()); in CreateFromProto()
691 proto.indices_are_sorted(), proto.unique_indices()); in CreateFromProto()
1618 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in CreateGather() argument
1621 indices_are_sorted); in CreateGather()
1628 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in CreateScatter() argument
1632 scatter_dim_numbers, indices_are_sorted, unique_indices); in CreateScatter()
Dhlo_parser.cc2456 optional<bool> indices_are_sorted = false; in ParseInstructionRhs() local
2458 &indices_are_sorted}; in ParseInstructionRhs()
2482 dim_numbers, *slice_sizes, indices_are_sorted.value())); in ParseInstructionRhs()
2503 optional<bool> indices_are_sorted = false; in ParseInstructionRhs() local
2505 &indices_are_sorted}; in ParseInstructionRhs()
2536 indices_are_sorted.value(), unique_indices.value())); in ParseInstructionRhs()
Dhlo_instruction.h947 absl::Span<const int64> slice_sizes, bool indices_are_sorted);
954 bool indices_are_sorted, bool unique_indices);
Dhlo_parser_test.cc926 …rt_index_map={0,1,2,3,4}, index_vector_dim=4, slice_sizes={30,29,28,27,26}, indices_are_sorted=true in CreateTestCases()
964 …={}, scatter_dims_to_operand_dims={0,1,2,3,4}, index_vector_dim=4, indices_are_sorted=true, to_app… in CreateTestCases()
Dhlo.proto245 bool indices_are_sorted = 67; field
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h849 bool indices_are_sorted = false);
854 absl::Span<const int64> slice_sizes, bool indices_are_sorted);
859 bool indices_are_sorted = false, bool unique_indices = false);
864 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted,
1395 bool indices_are_sorted);
1399 bool indices_are_sorted, bool unique_indices);
2444 bool indices_are_sorted = false);
2450 bool indices_are_sorted = false, bool unique_indices = false);
Dxla_builder.cc2273 bool indices_are_sorted) { in Gather() argument
2282 slice_sizes, indices_are_sorted); in Gather()
2289 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in GatherInternal() argument
2291 instr.set_indices_are_sorted(indices_are_sorted); in GatherInternal()
2305 bool indices_are_sorted, bool unique_indices) { in Scatter() argument
2319 indices_are_sorted, unique_indices); in Scatter()
2326 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal() argument
2330 instr.set_indices_are_sorted(indices_are_sorted); in ScatterInternal()
4877 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in Gather() argument
4879 slice_sizes, indices_are_sorted); in Gather()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc594 Cast<const HloScatterInstruction>(use)->indices_are_sorted(); in CreateLiteralForConstrainedUses()
597 Cast<const HloGatherInstruction>(use)->indices_are_sorted(); in CreateLiteralForConstrainedUses()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir492 // CHECK-SAME: indices_are_sorted=true
493 …: tensor<1xi64>, start_index_map = dense<[0, 1]> : tensor<2xi64>}, indices_are_sorted = true, slic…
769 indices_are_sorted = true,
781 …ims={0,1}, scatter_dims_to_operand_dims={0,1}, index_vector_dim=1, indices_are_sorted=true, unique…
Dimport.hlotxt421 // CHECK-SAME: indices_are_sorted = true
428 indices_are_sorted=true,
778 // CHECK-SAME: indices_are_sorted = false
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc360 builder_->getBoolAttr(gather_instruction->indices_are_sorted()))); in ImportInstructionImpl()
438 builder_->getBoolAttr(scatter->indices_are_sorted()))); in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_patterns.td748 $dimension_numbers, $indices_are_sorted),
752 $indices_are_sorted),
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlegalize_hlo.mlir1896 …se<> : tensor<0xi64>, start_index_map = dense<0> : tensor<1xi64>}, indices_are_sorted = false, sli…
1909 …e<2> : tensor<1xi64>, start_index_map = dense<0> : tensor<1xi64>}, indices_are_sorted = false, sli…
1953 indices_are_sorted = false,
1976 indices_are_sorted = false,
1999 indices_are_sorted = false,
2022 indices_are_sorted = false,
2045 indices_are_sorted = false,
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td1020 DefaultValuedAttr<BoolAttr, "false">:$indices_are_sorted
1093 DefaultValuedAttr<BoolAttr, "false">:$indices_are_sorted,
Dlhlo_ops.td473 DefaultValuedAttr<BoolAttr, "false">:$indices_are_sorted,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir203 // CHECK-SAME: indices_are_sorted = false

12