Searched defs:indices_are_sorted (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 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()
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()
|
| D | hlo.proto | 245 bool indices_are_sorted = 67; field
|
| D | hlo_parser.cc | 2456 optional<bool> indices_are_sorted = false; in ParseInstructionRhs() local
2503 optional<bool> indices_are_sorted = false; in ParseInstructionRhs() local
|
| D | hlo_instruction.cc | 1618 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in CreateGather()
1628 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in CreateScatter()
|
| D | hlo_instructions.cc | 2726 absl::Span<const int64> slice_sizes, bool indices_are_sorted) in HloGatherInstruction()
2820 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in HloScatterInstruction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 483 indices_are_sorted=False, name=None): argument
494 dimension_numbers, indices_are_sorted=False, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 290 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in GatherInternal()
303 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2273 bool indices_are_sorted) { in Gather()
2289 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in GatherInternal()
2305 bool indices_are_sorted, bool unique_indices) { in Scatter()
2326 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
4877 absl::Span<const int64> slice_sizes, bool indices_are_sorted) { in Gather()
4885 bool indices_are_sorted, bool unique_indices) { in Scatter()
|