/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
D | hlo-legalize-gather-to-torch-index-select.mlir | 10 …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…
|
D | canonicalize.mlir | 1023 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/ |
D | xla.py | 483 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/ |
D | gpu_ops.mlir | 13 // CHECK: indices_are_sorted = false 29 indices_are_sorted = false,
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | gather_scatter_handler.cc | 201 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/ |
D | mlir_hlo_builder.h | 182 absl::Span<const int64> slice_sizes, bool indices_are_sorted) override; 187 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted,
|
D | mlir_hlo_builder.cc | 290 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/ |
D | hlo_instructions.cc | 2726 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 …]
|
D | hlo_instructions.h | 1658 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()
|
D | hlo_instruction.cc | 676 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()
|
D | hlo_parser.cc | 2456 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()
|
D | hlo_instruction.h | 947 absl::Span<const int64> slice_sizes, bool indices_are_sorted); 954 bool indices_are_sorted, bool unique_indices);
|
D | hlo_parser_test.cc | 926 …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()
|
D | hlo.proto | 245 bool indices_are_sorted = 67; field
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 849 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);
|
D | xla_builder.cc | 2273 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/ |
D | test_utils.cc | 594 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/ |
D | export.mlir | 492 // 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…
|
D | import.hlotxt | 421 // CHECK-SAME: indices_are_sorted = true 428 indices_are_sorted=true, 778 // CHECK-SAME: indices_are_sorted = false
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | hlo_function_importer.cc | 360 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/ |
D | legalize_tf_patterns.td | 748 $dimension_numbers, $indices_are_sorted), 752 $indices_are_sorted),
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
D | legalize_hlo.mlir | 1896 …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/ |
D | hlo_ops.td | 1020 DefaultValuedAttr<BoolAttr, "false">:$indices_are_sorted 1093 DefaultValuedAttr<BoolAttr, "false">:$indices_are_sorted,
|
D | lhlo_ops.td | 473 DefaultValuedAttr<BoolAttr, "false">:$indices_are_sorted,
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
D | legalize-tf-with-tf2xla.mlir | 203 // CHECK-SAME: indices_are_sorted = false
|