Home
last modified time | relevance | path

Searched refs:reduce_on_cols (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtopk_op.cc114 typename Eigen::IndexList<Eigen::type2index<1>> reduce_on_cols; in Compute() local
118 Eigen::array<int, 1> reduce_on_cols = {1}; in Compute()
123 input.maximum(/*dims=*/reduce_on_cols).eval().reshape(rows_by_one); in Compute()