Searched refs:dimension_to_sort_bound (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 147 int64 dimension_to_sort_bound, absl::Span<const int64> xor_masks, in EmitTiledCompareLoop() argument
172 dimension_to_sort_bound)), in EmitTiledCompareLoop()
182 dimension_to_sort_bound)), in EmitTiledCompareLoop()
235 if (dimension_to_sort_bound % tile_size) { in EmitTiledCompareLoop()
244 RoundDownToNearest(dimension_to_sort_bound, tile_size))), in EmitTiledCompareLoop()
247 dimension_to_sort_bound % tile_size, params.size(), in EmitTiledCompareLoop()
311 int64 dimension_to_sort_bound = keys_shape.dimensions(dimension_to_sort); in EmitSortInPlace() local
365 keys_index, dimension_to_sort, dimension_to_sort_bound, xor_masks, in EmitSortInPlace()
385 dimension_to_sort_bound, values_arrays.size(), tiles_index[rank - 1], in EmitSortInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2890 uint64 dimension_to_sort_bound = keys_shape.dimensions(dimension_to_sort); in EmitSortFromMlir() local
2891 int64 num_stages = tensorflow::Log2Ceiling(dimension_to_sort_bound); in EmitSortFromMlir()
2893 CHECK_GE(1ULL << num_stages, dimension_to_sort_bound); in EmitSortFromMlir()
2894 CHECK_LT(1ULL << (num_stages - 1), dimension_to_sort_bound); in EmitSortFromMlir()
2938 int64 rounded_bound = RoundUpToNearest(dimension_to_sort_bound, kTileSize); in EmitSortFromMlir()
|