Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtopk_rewriter.cc138 const int64_t sort_dim = sort->sort_dimension(); in SortIsInTopK() local
206 const int64_t sort_dim = sort->sort_dimension(); in TransformToCustomCall() local
Ddynamic_padder.cc1347 int64_t sort_dim = sort->sort_dimension(); in RewriteDynamicSort() local
Dhlo_evaluator.cc3480 int64_t sort_dim = sort->dimensions(0); in HandleSort() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dcustom_call_handler.cc83 const int64_t sort_dim = 1; in HandleCustomCallTopK() local
Dspmd_partitioner_util.cc1135 const int64_t sort_dim = sort->sort_dimension(); in GetKValueInTopKWhenPartitionSortDim() local
Dspmd_partitioner.cc2468 const int64_t sort_dim = sort->sort_dimension(); in HandleSort() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1592 auto sort_dim = op.dimension(); in matchAndRewrite() local