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 | 146 const IrArray::Index& tiled_keys_index, int64 dimension_to_sort, in EmitTiledCompareLoop() argument
165 b->CreateShl(tiled_keys_index[dimension_to_sort], value_one); in EmitTiledCompareLoop()
194 keys_multi_index[dimension_to_sort] = index; in EmitTiledCompareLoop()
241 b->CreateMul(tiled_keys_index[dimension_to_sort], in EmitTiledCompareLoop()
273 keys_multi_index[dimension_to_sort] = index; in EmitTiledCompareLoop()
295 int64 dimension_to_sort, const std::vector<IrArray>& values_arrays, in EmitSortInPlace() argument
311 int64 dimension_to_sort_bound = keys_shape.dimensions(dimension_to_sort); in EmitSortInPlace()
316 if (dimension != dimension_to_sort) { in EmitSortInPlace()
322 iteration_order_to_logical_order[dim] = dimension_to_sort; in EmitSortInPlace()
365 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/gpu/ |
| D | ir_emitter.h | 206 void EmitCompareLoop(int64 dimension_to_sort,
|
| D | ir_emitter_unnested.cc | 2863 int64 dimension_to_sort = sort_op.dimension(); in EmitSortFromMlir() local
2890 uint64 dimension_to_sort_bound = keys_shape.dimensions(dimension_to_sort); in EmitSortFromMlir()
2929 standard_iteration_shape.set_dimensions(dimension_to_sort, in EmitSortFromMlir()
2943 iteration_shape.set_dimensions(dimension_to_sort, num_iterations_in_sort_dim); in EmitSortFromMlir()
3008 dimension_to_sort, values_arrays, IrName(context.name), xor_masks, &b_, in EmitSortFromMlir()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.h | 181 int64 dimension_to_sort, bool is_stable, HloComputation::Builder* builder,
|
| D | hlo_creation_utils.cc | 418 int64 dimension_to_sort, bool is_stable, HloComputation::Builder* builder, in MakeSortHlo() argument
436 sort_shape, dimension_to_sort, operands, compare_computation, is_stable)); in MakeSortHlo()
|
| D | algebraic_simplifier.cc | 5029 int64 dimension_to_sort = sort->dimensions(0); in HandleSort() local
5031 operand->shape().dimensions(dimension_to_sort) <= 1) { in HandleSort()
|