Searched refs:sort_dimension_elements (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_key_value_sort.cc | 48 tensorflow::int64 sort_dimension_elements = b; in __xla_cpu_runtime_KeyValueSort() local
52 std::unique_ptr<tensorflow::int64[]> indices(new tensorflow::int64[sort_dimension_elements]); in __xla_cpu_runtime_KeyValueSort()
54 std::iota(indices.get(), indices.get() + sort_dimension_elements, 0); in __xla_cpu_runtime_KeyValueSort()
56 new std::string[sort_dimension_elements]); in __xla_cpu_runtime_KeyValueSort()
61 std::iota(indices.get(), indices.get() + sort_dimension_elements, 0); 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 (tensorflow::int64 i = 0; i < sort_dimension_elements; ++i) { in __xla_cpu_runtime_KeyValueSort()
105 for (tensorflow::int64 i = 0; i < sort_dimension_elements; ++i) { in __xla_cpu_runtime_KeyValueSort()
|
| D | ir_emitter.cc | 589 int64 sort_dimension_elements = in HandleSort() local
624 {b_.getInt64(higher_dimensions), b_.getInt64(sort_dimension_elements), in HandleSort()
|