Home
last modified time | relevance | path

Searched defs:dimension_to_sort (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc145 const IrArray::Index& tiled_keys_index, int64 dimension_to_sort, in EmitTiledCompareLoop()
293 int64 dimension_to_sort, const std::vector<IrArray>& values_arrays, in EmitSortInPlace()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc290 int64 dimension_to_sort, bool is_stable, HloComputation::Builder* builder, in MakeSortHlo()
Dalgebraic_simplifier.cc3326 int64 dimension_to_sort = sort->dimensions(0); in HandleSort() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1212 int64 dimension_to_sort = sort->dimensions(0); in HandleSort() local