| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_ref.cpp | 22 TensorRef<Tensor<int, 1>> ref3(input); in test_simple_lvalue_ref()
23 TensorRef<Tensor<int, 1>> ref4 = input; in test_simple_lvalue_ref()
55 TensorRef<Tensor<int, 1>> ref3(input1 + input2); in test_simple_rvalue_ref()
56 TensorRef<Tensor<int, 1>> ref4 = input1 + input2; in test_simple_rvalue_ref()
75 TensorRef<Tensor<float, 3>> ref(input); in test_multiple_dims()
98 TensorRef<Tensor<float, 5>> slice = tensor.slice(indices, sizes); in test_slice()
124 TensorRef<Tensor<float, 3>> ref(input); in test_ref_of_ref()
125 TensorRef<Tensor<float, 3>> ref_of_ref(ref); in test_ref_of_ref()
126 TensorRef<Tensor<float, 3>> ref_of_ref2; in test_ref_of_ref()
154 TensorRef<Tensor<float, 3>> input_ref(input); in test_ref_in_expr()
[all …]
|
| D | cxx11_tensor_custom_op.cpp | 97 TensorRef<Tensor<float, 2> > val = result.chip<2>(i); in test_custom_binary_op()
|
| D | cxx11_tensor_morphing.cpp | 125 TensorRef<Tensor<const float, 1> > slice_ref(m.slice(offsets, m.dimensions())); in test_const_slice()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorRef.h | 119 template<typename PlainObjectType> class TensorRef : public TensorBase<TensorRef<PlainObjectType> >
122 typedef TensorRef<PlainObjectType> Self;
144 EIGEN_STRONG_INLINE TensorRef() : m_evaluator(NULL) { in TensorRef() function
148 …EIGEN_STRONG_INLINE TensorRef(const Expression& expr) : m_evaluator(new internal::TensorLazyEvalua… in TensorRef() function
153 EIGEN_STRONG_INLINE TensorRef& operator = (const Expression& expr) {
160 ~TensorRef() { in ~TensorRef()
164 TensorRef(const TensorRef& other) : m_evaluator(other.m_evaluator) { in TensorRef() function
169 TensorRef& operator = (const TensorRef& other) {
356 struct TensorEvaluator<const TensorRef<Derived>, Device>
367 Layout = TensorRef<Derived>::Layout,
[all …]
|
| D | TensorTraits.h | 105 struct traits<TensorRef<PlainObjectType> >
158 struct eval<TensorRef<PlainObjectType>, Eigen::Dense>
160 typedef const TensorRef<PlainObjectType>& type;
164 struct eval<const TensorRef<PlainObjectType>, Eigen::Dense>
166 typedef const TensorRef<PlainObjectType>& type;
213 struct nested<TensorRef<PlainObjectType> >
215 typedef const TensorRef<PlainObjectType>& type;
219 struct nested<const TensorRef<PlainObjectType> >
221 typedef const TensorRef<PlainObjectType>& type;
|
| D | README.md | 90 #### Class TensorRef
92 See Assigning to a TensorRef below.
255 * Assignment to a TensorRef.
358 #### Assigning to a TensorRef.
361 can avoid materializing the value in a full tensor by using a TensorRef.
363 A TensorRef is a small wrapper class for any Eigen Operation. It provides
365 the expression. TensorRef is convenient, because the Operation themselves do
368 // Create a TensorRef for the expression. The expression is not
370 TensorRef<Tensor<float, 3> > ref = ((t1 + t2) * 0.2f).exp();
377 Only use TensorRef when you need a subset of the values of the expression.
[all …]
|
| D | TensorForwardDeclarations.h | 28 template<typename PlainObjectType> class TensorRef; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_dense_binary_op_shared.cc | 48 using Eigen::TensorRef;
124 TensorRef<Eigen::Tensor<const T, NDIM, Eigen::RowMajor>> rhs_ref = \ in Compute()
|
| D | eigen_backward_spatial_convolutions.h | 106 TensorRef<Tensor<typename internal::traits<Kernel>::Scalar,
110 TensorRef<Tensor<OutScalar, internal::traits<OutputBackward>::NumDimensions,
356 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
360 TensorRef<Tensor<OutScalar, internal::traits<OutputBackward>::NumDimensions,
|
| D | eigen_pooling.h | 72 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
189 TensorRef<Tensor<typename internal::traits<Input>::Scalar, in CuboidMaxPooling()
429 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
545 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
|
| D | eigen_backward_cuboid_convolutions.h | 95 const TensorRef<const Tensor<typename internal::traits<Kernel>::Scalar,
99 const TensorRef<
397 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
401 TensorRef<Tensor<typename internal::traits<OutputBackward>::Scalar,
|
| D | eigen_attention.h | 70 typedef TensorRef<Tensor<typename internal::traits<Input>::Scalar, 4, in dimensions()
88 typedef TensorRef<Tensor<typename internal::traits<Input>::Scalar, 4, in eval()
|
| D | eigen_spatial_convolutions-inl.h | 1365 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
1369 TensorRef<Tensor<typename internal::traits<Kernel>::Scalar,
|
| D | eigen_cuboid_convolution.h | 1651 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
1655 TensorRef<Tensor<typename internal::traits<Kernel>::Scalar,
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 104 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
108 TensorRef<Tensor<typename internal::traits<Kernel>::Scalar,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 101 TensorRef<Tensor<typename internal::traits<Input>::Scalar,
105 TensorRef<Tensor<typename internal::traits<Kernel>::Scalar,
|
| /external/eigen/unsupported/Eigen/CXX11/ |
| D | Tensor | 146 #include "src/Tensor/TensorRef.h"
|