Searched refs:sort_dimension_offset (Results 1 – 1 of 1) sorted by relevance
55 int64 sort_dimension_offset = c; in __xla_cpu_runtime_KeyValueSort() local70 index % sort_dimension_offset + in __xla_cpu_runtime_KeyValueSort()71 (index - index % sort_dimension_offset) * sort_dimension_elements; in __xla_cpu_runtime_KeyValueSort()73 int64 memory_index_lhs = (base_offset + a * sort_dimension_offset) * in __xla_cpu_runtime_KeyValueSort()75 int64 memory_index_rhs = (base_offset + b * sort_dimension_offset) * in __xla_cpu_runtime_KeyValueSort()98 (base_offset + indices[i] * sort_dimension_offset) * in __xla_cpu_runtime_KeyValueSort()106 int64 memory_index = (base_offset + i * sort_dimension_offset) * in __xla_cpu_runtime_KeyValueSort()