diff options
| author |  Michael Kuperstein <mkuper@google.com> | 2018-10-09 19:41:35 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 19:46:05 -0700 |
| commit | 58fcfc98cd59ae3952399fc55380b8733df08df9 (patch) | |
| tree | 24d5ac5d6691e73c227f5afa5ef68ba2ecba4ec0 /tensorflow/compiler/xla/service/hlo_instruction.h | |
| parent | 93eef55c4d04af24a6c8080f34629db179634f07 (diff) | |
[XLA] Add documentation and HLO-level support for multi-value sort.
No support in any of the backends, and not yet exposed through XlaBuilder.
PiperOrigin-RevId: 216465753
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_instruction.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_instruction.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_instruction.h b/tensorflow/compiler/xla/service/hlo_instruction.h index 44f776ebac..93ff04b1e4 100644 --- a/tensorflow/compiler/xla/service/hlo_instruction.h +++ b/tensorflow/compiler/xla/service/hlo_instruction.h @@ -670,10 +670,10 @@ class HloInstruction { const Shape& shape, HloInstruction* operand, absl::Span<const int64> dimensions); - // Creates a sort op, with a keys operand, and an optional values operand. + // Creates a sort op, with a keys operand, and optional values operands. static std::unique_ptr<HloInstruction> CreateSort( const Shape& shape, int64 dimension, HloInstruction* keys, - HloInstruction* values = nullptr); + absl::Span<HloInstruction* const> values = {}); // Creates a while instruction, given a condition computation, a body // computation, and the initial value for the input of the computations. For |