Searched refs:return_dim (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorArgMax.h | 175 const int return_dim,
177 : m_xpr(expr), m_reduce_op(reduce_op), m_return_dim(return_dim), m_reduce_dims(reduce_dims) {}
190 int return_dim() const { return m_return_dim; }
225 m_return_dim(op.return_dim()) {
|
| D | TensorBase.h | 640 argmax(const int return_dim) const { in argmax() argument
642 in_dims[0] = return_dim; in argmax()
646 …d>(derived(), internal::ArgMaxTupleReducer<Tuple<Index, CoeffReturnType> >(), return_dim, in_dims); in argmax()
653 argmin(const int return_dim) const { in argmin() argument
655 in_dims[0] = return_dim; in argmin()
659 …d>(derived(), internal::ArgMinTupleReducer<Tuple<Index, CoeffReturnType> >(), return_dim, in_dims); in argmin()
|