Searched defs:indices_are_sorted (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 1767 bool indices_are_sorted() const { return indices_are_sorted_; } in indices_are_sorted() function
1768 void set_indices_are_sorted(bool indices_are_sorted) { in set_indices_are_sorted()
1811 bool indices_are_sorted() const { return indices_are_sorted_; } in indices_are_sorted() function
1812 void set_indices_are_sorted(bool indices_are_sorted) { in set_indices_are_sorted()
|
| D | hlo.proto | 284 bool indices_are_sorted = 67; field
|
| D | hlo_parser.cc | 2612 optional<bool> indices_are_sorted = false; in ParseInstructionRhs() local
2659 optional<bool> indices_are_sorted = false; in ParseInstructionRhs() local
|
| D | hlo_instruction.cc | 1805 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in CreateGather()
1815 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in CreateScatter()
|
| D | hlo_instructions.cc | 2890 absl::Span<const int64> slice_sizes, bool indices_are_sorted) in HloGatherInstruction()
2984 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in HloScatterInstruction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 551 indices_are_sorted=False, name=None): argument
562 dimension_numbers, indices_are_sorted=False, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 307 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in GatherInternal()
320 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2301 bool indices_are_sorted) { in Gather()
2317 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in GatherInternal()
2333 bool indices_are_sorted, bool unique_indices) { in Scatter()
2354 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
4739 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in Gather()
4747 bool indices_are_sorted, bool unique_indices) { in Scatter()
|