Home
last modified time | relevance | path

Searched defs:sort_dim (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtopk_rewriter.cc91 const int64 sort_dim = sort->sort_dimension(); in Run() local
Ddynamic_padder.cc1198 int64 sort_dim = sort->sort_dimension(); in RewriteDynamicSort() local
Dhlo_evaluator.cc2180 int64 sort_dim = sort->dimensions(0); in HandleSort() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1658 const int64 sort_dim = sort->sort_dimension(); in HandleSort() local
1815 const int64 sort_dim = 1; in HandleCustomCall() local
Dspmd_partitioner_util.cc1173 const int64 sort_dim = sort->sort_dimension(); in GetKValueInTopKWhenPartitionSortDim() local