Searched refs:dimension_to_sort (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 145 const IrArray::Index& tiled_keys_index, int64 dimension_to_sort, in EmitTiledCompareLoop() argument
164 b->CreateShl(tiled_keys_index[dimension_to_sort], value_one); in EmitTiledCompareLoop()
193 keys_multi_index[dimension_to_sort] = index; in EmitTiledCompareLoop()
240 b->CreateMul(tiled_keys_index[dimension_to_sort], in EmitTiledCompareLoop()
271 keys_multi_index[dimension_to_sort] = index; in EmitTiledCompareLoop()
293 int64 dimension_to_sort, const std::vector<IrArray>& values_arrays, in EmitSortInPlace() argument
309 int64 dimension_to_sort_bound = keys_shape.dimensions(dimension_to_sort); in EmitSortInPlace()
314 if (dimension != dimension_to_sort) { in EmitSortInPlace()
320 iteration_order_to_logical_order[dim] = dimension_to_sort; in EmitSortInPlace()
363 keys_index, dimension_to_sort, dimension_to_sort_bound, xor_masks, in EmitSortInPlace()
[all …]
|
| D | sort_util.h | 38 int64 dimension_to_sort, const std::vector<IrArray>& values_arrays,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.h | 138 int64 dimension_to_sort, bool is_stable, HloComputation::Builder* builder,
|
| D | hlo_creation_utils.cc | 290 int64 dimension_to_sort, bool is_stable, HloComputation::Builder* builder, in MakeSortHlo() argument
308 sort_shape, dimension_to_sort, operands, compare_computation, is_stable)); in MakeSortHlo()
|
| D | algebraic_simplifier.cc | 3326 int64 dimension_to_sort = sort->dimensions(0); in HandleSort() local
3328 operand->shape().dimensions(dimension_to_sort) <= 1) { in HandleSort()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.h | 219 void EmitCompareLoop(int64 dimension_to_sort,
|
| D | ir_emitter_unnested.cc | 1212 int64 dimension_to_sort = sort->dimensions(0); in HandleSort() local
1238 uint64 dimension_to_sort_bound = keys_shape.dimensions(dimension_to_sort); in HandleSort()
1276 standard_iteration_shape.set_dimensions(dimension_to_sort, in HandleSort()
1290 iteration_shape.set_dimensions(dimension_to_sort, num_iterations_in_sort_dim); in HandleSort()
1334 dimension_to_sort, values_arrays, IrName(sort), xor_masks, &b_, in HandleSort()
|