TensorArgMax.h | 174 const ReduceOp& reduce_op, 177 : m_xpr(expr), m_reduce_op(reduce_op), m_return_dim(return_dim), m_reduce_dims(reduce_dims) {} 184 const ReduceOp& reduce_op() const { return m_reduce_op; } function in class:Eigen::TensorTupleReducerOp 224 m_impl(op.expression().index_tuples().reduce(op.reduce_dims(), op.reduce_op()), device),
|