Searched defs:indices_are_sorted (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instructions.h | 2113 bool indices_are_sorted() const { return indices_are_sorted_; } in indices_are_sorted() function
2114 void set_indices_are_sorted(bool indices_are_sorted) { in set_indices_are_sorted()
2160 bool indices_are_sorted() const { return indices_are_sorted_; } in indices_are_sorted() function
2161 void set_indices_are_sorted(bool indices_are_sorted) { in set_indices_are_sorted()
|
| H A D | hlo.proto | 304 bool indices_are_sorted = 67; field
|
| H A D | hlo_instruction.cc | 1929 absl::Span<const int64_t> slice_sizes, bool indices_are_sorted) { in CreateGather()
1939 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in CreateScatter()
1950 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in CreateScatter()
|
| H A D | hlo_parser.cc | 2819 optional<bool> indices_are_sorted = false; in CreateInstruction() local
2864 optional<bool> indices_are_sorted = false; in CreateInstruction() local
|
| H A D | hlo_instructions.cc | 3050 absl::Span<const int64_t> slice_sizes, bool indices_are_sorted) in HloGatherInstruction()
3143 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in HloScatterInstruction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| H A D | xla.py | 572 indices_are_sorted=False, name=None): argument
583 dimension_numbers, indices_are_sorted=False, name=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 384 absl::Span<const int64_t> slice_sizes, bool indices_are_sorted) { in GatherInternal()
397 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 2398 bool indices_are_sorted) { in Gather()
2414 absl::Span<const int64_t> slice_sizes, bool indices_are_sorted) { in GatherInternal()
2430 bool indices_are_sorted, bool unique_indices) { in Scatter()
2440 bool indices_are_sorted, bool unique_indices) { in Scatter()
2477 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
4901 absl::Span<const int64_t> slice_sizes, bool indices_are_sorted) { in Gather()
4909 bool indices_are_sorted, bool unique_indices) { in Scatter()
4919 bool indices_are_sorted, bool unique_indices) { in Scatter()
|