diff options
| author |  Michael Kuperstein <mkuper@google.com> | 2018-07-10 13:04:30 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-10 13:08:18 -0700 |
| commit | 1882427291f212cd04d09f2b35af4a78f6d771b5 (patch) | |
| tree | c6172cec133d7f7fed37757db67822e074d658a9 /tensorflow/compiler/xla/service/hlo_instruction.h | |
| parent | ccbbd4484a29fb3ee7d0d67abcebafdb48c9059b (diff) | |
[XLA] Generalize sort semantics to Rk.
PiperOrigin-RevId: 203997296
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_instruction.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_instruction.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_instruction.h b/tensorflow/compiler/xla/service/hlo_instruction.h index dac5f1cf86..b392d65636 100644 --- a/tensorflow/compiler/xla/service/hlo_instruction.h +++ b/tensorflow/compiler/xla/service/hlo_instruction.h @@ -615,7 +615,7 @@ class HloInstruction { // Creates a sort op, with a keys operand, and an optional values operand. static std::unique_ptr<HloInstruction> CreateSort( - const Shape& shape, HloInstruction* keys, + const Shape& shape, int64 dimension, HloInstruction* keys, HloInstruction* values = nullptr); // Creates a while instruction, given a condition computation, a body |