Home
last modified time | relevance | path

Searched refs:iota_index (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dstable_sort_expander.cc44 int64 iota_index = -1; in ExpandInstruction() local
56 iota_index = sort->operand_index(operand); in ExpandInstruction()
63 if (iota_index == -1) { in ExpandInstruction()
126 iota_index = sort->operand_count() - 1; in ExpandInstruction()
187 scalar_pred, comparator->parameter_instruction(2 * iota_index), in ExpandInstruction()
188 comparator->parameter_instruction(2 * iota_index + 1), in ExpandInstruction()