Searched refs:index_tuples (Results 1 – 3 of 3) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_argmax.cpp | 26 Tensor<Tuple<DenseIndex, float>, 4, DataLayout> index_tuples(2,3,5,7); in test_simple_index_tuples() local
27 index_tuples = tensor.index_tuples(); in test_simple_index_tuples()
30 const Tuple<DenseIndex, float>& v = index_tuples.coeff(n); in test_simple_index_tuples()
43 Tensor<Tuple<DenseIndex, float>, 4, DataLayout> index_tuples(2,3,5,7); in test_index_tuples_dim() local
45 index_tuples = tensor.index_tuples(); in test_index_tuples_dim()
48 const Tuple<DenseIndex, float>& v = index_tuples(n); //(i, j, k, l); in test_index_tuples_dim()
61 Tensor<Tuple<DenseIndex, float>, 4, DataLayout> index_tuples(2,3,5,7); in test_argmax_tuple_reducer() local
62 index_tuples = tensor.index_tuples(); in test_argmax_tuple_reducer()
66 reduced = index_tuples.reduce( in test_argmax_tuple_reducer()
76 reduced_by_dims = index_tuples.reduce( in test_argmax_tuple_reducer()
[all …]
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorArgMax.h | 224 m_impl(op.expression().index_tuples().reduce(op.reduce_dims(), op.reduce_op()), device),
|
| D | TensorBase.h | 796 index_tuples() const { in index_tuples() function
|