Searched refs:sort_dimension_elements (Results 1 – 2 of 2) sorted by relevance
53 int64 sort_dimension_elements = b; in __xla_cpu_runtime_KeyValueSort() local57 std::unique_ptr<int64[]> indices(new int64[sort_dimension_elements]); in __xla_cpu_runtime_KeyValueSort()59 std::iota(indices.get(), indices.get() + sort_dimension_elements, 0); in __xla_cpu_runtime_KeyValueSort()61 new std::string[sort_dimension_elements]); in __xla_cpu_runtime_KeyValueSort()71 (index - index % sort_dimension_offset) * sort_dimension_elements; in __xla_cpu_runtime_KeyValueSort()87 std::stable_sort(indices.get(), indices.get() + sort_dimension_elements, in __xla_cpu_runtime_KeyValueSort()90 std::sort(indices.get(), indices.get() + sort_dimension_elements, in __xla_cpu_runtime_KeyValueSort()96 for (int64 i = 0; i < sort_dimension_elements; ++i) { in __xla_cpu_runtime_KeyValueSort()105 for (int64 i = 0; i < sort_dimension_elements; ++i) { in __xla_cpu_runtime_KeyValueSort()
564 int64 sort_dimension_elements = in HandleSort() local612 {b_.getInt64(higher_dimensions), b_.getInt64(sort_dimension_elements), in HandleSort()