| /external/tensorflow/tensorflow/core/framework/ |
| tensor_types.h | 27 typedef Eigen::TensorMap<Eigen::Tensor<T, NDIMS, Eigen::RowMajor, IndexType>,
30 typedef Eigen::TensorMap<
35 typedef Eigen::TensorMap<Eigen::Tensor<T, NDIMS, Eigen::RowMajor, IndexType> >
37 typedef Eigen::TensorMap<
41 typedef Eigen::TensorMap<Eigen::Tensor<T, NDIMS, Eigen::RowMajor, int>,
46 typedef Eigen::TensorMap<
50 typedef Eigen::TensorMap<Eigen::TensorFixedSize<const T, Eigen::Sizes<>,
56 typedef Eigen::TensorMap<
59 typedef Eigen::TensorMap<Eigen::TensorFixedSize<const T, Eigen::Sizes<>,
64 typedef Eigen::TensorMap<Eigen::Tensor<T, 1, Eigen::RowMajor, IndexType>
[all...] |
| /external/eigen/unsupported/test/ |
| cxx11_tensor_const.cpp | 21 TensorMap<Tensor<const int, 3> > constant(random.data(), 2, 3, 7);
40 TensorMap<Tensor<const int, 3> > constant1(random.data(), 2, 3, 7);
41 TensorMap<const Tensor<int, 3> > constant2(random.data(), 2, 3, 7);
42 const TensorMap<Tensor<int, 3> > constant3(random.data(), 2, 3, 7);
|
| cxx11_tensor_of_const_values.cpp | 20 TensorMap<Tensor<const float, 2>> mat1(data1, 2, 3);
22 const TensorMap<Tensor<float, 2>> mat2(data2, 2, 3);
56 TensorMap<Tensor<const float, 2>> mat1(data1, 2, 3);
58 TensorMap<Tensor<float, 2>> mat2(data2, 2, 3);
82 TensorMap<Tensor<const float, 2>> mat1(data1, 2, 3);
84 TensorMap<Tensor<float, 2>> mat2(data2, 2, 3);
|
| cxx11_tensor_map.cpp | 22 TensorMap<Tensor<const int, 0> > scalar3(scalar1.data());
23 TensorMap<Tensor<const int, 0, RowMajor> > scalar4(scalar2.data());
40 TensorMap<Tensor<const int, 1> > vec3(vec1.data(), 6);
41 TensorMap<Tensor<const int, 1, RowMajor> > vec4(vec2.data(), 6);
88 TensorMap<Tensor<const int, 2> > mat3(mat1.data(), 2, 3);
89 TensorMap<Tensor<const int, 2, RowMajor> > mat4(mat2.data(), 2, 3);
132 TensorMap<Tensor<const int, 3> > mat3(mat1.data(), 2, 3, 7);
133 TensorMap<Tensor<const int, 3, RowMajor> > mat4(mat2.data(), 2, 3, 7);
176 TensorMap<Tensor<int, 3> > mat3(mat1);
177 TensorMap<Tensor<int, 3, RowMajor> > mat4(mat2)
[all...] |
| cxx11_tensor_reduction_sycl.cpp | 42 TensorMap<Tensor<float, 2> > in_gpu(gpu_in_data, tensorRange);
43 TensorMap<Tensor<float, 0> > out_gpu(gpu_out_data);
77 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange);
78 TensorMap<Tensor<float, 2> > out_gpu(gpu_out_data, reduced_tensorRange);
115 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange);
116 TensorMap<Tensor<float, 2> > out_gpu(gpu_out_data, reduced_tensorRange);
|
| cxx11_tensor_broadcast_sycl.cpp | 26 using Eigen::TensorMap;
50 TensorMap<Tensor<float, 4>> gpu_in(gpu_in_data, in_range);
51 TensorMap<Tensor<float, 4>> gpu_out(gpu_out_data, out_range);
|
| cxx11_tensor_forced_eval.cpp | 25 TensorMap<Tensor<float, 2> > mat1(m1.data(), 3,3);
26 TensorMap<Tensor<float, 2> > mat2(m2.data(), 3,3);
64 const TensorMap<Tensor<const float, 2> > input_tensor(input.data(), 3, 3);
|
| cxx11_tensor_mixed_indices.cpp | 28 TensorMap<Tensor<float, 1, ColMajor>> vec3(data3, 6);
31 TensorMap<Tensor<float, 1, ColMajor, int>> vec4(data4, 6);
|
| cxx11_tensor_forced_eval_sycl.cpp | 42 // creating TensorMap from tensor
43 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_in1(gpu_in1_data, tensorRange);
44 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_in2(gpu_in2_data, tensorRange);
45 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_out(gpu_out_data, tensorRange);
|
| cxx11_tensor_assign.cpp | 32 TensorMap<Tensor<int, 1> > vec3(col_major, 6);
33 TensorMap<Tensor<int, 1, RowMajor> > vec4(row_major, 6);
95 TensorMap<Tensor<int, 2> > mat3(row_major, 2, 3);
96 TensorMap<Tensor<int, 2, RowMajor> > mat4(col_major, 2, 3);
155 TensorMap<Tensor<int, 3> > mat3(col_major, 2, 3, 7);
156 TensorMap<Tensor<int, 3, RowMajor> > mat4(row_major, 2, 3, 7);
219 TensorMap<Tensor<int, 1> > orig_map(orig, 5);
220 TensorMap<Tensor<int, 1> > dest_map(dest, 5);
|
| cxx11_tensor_sycl.cpp | 28 using Eigen::TensorMap;
49 TensorMap<Tensor<float, 3>> gpu_in1(gpu_in1_data, tensorRange);
50 TensorMap<Tensor<float, 3>> gpu_in2(gpu_in2_data, tensorRange);
51 TensorMap<Tensor<float, 3>> gpu_in3(gpu_in3_data, tensorRange);
52 TensorMap<Tensor<float, 3>> gpu_out(gpu_out_data, tensorRange);
|
| /external/eigen/bench/tensors/ |
| tensor_benchmarks.h | 14 using Eigen::TensorMap;
56 const TensorMap<Tensor<int, 2, 0, TensorIndex>, Eigen::Aligned> A((int*)a_, sizes);
57 TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, sizes);
72 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes);
87 const TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, sizes);
88 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
89 TensorMap<Tensor<T, 2>, Eigen::Aligned> C(c_, sizes);
117 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size);
120 TensorMap<Tensor<T, 1, 0, TensorIndex>, Eigen::Aligned> C(c_, output_size);
134 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size)
[all...] |
| tensor_benchmarks_sycl.cc | 11 using Eigen::TensorMap;
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorTraits.h | 84 struct traits<TensorMap<PlainObjectType, Options_, MakePointer_> >
146 struct eval<TensorMap<PlainObjectType, Options, MakePointer>, Eigen::Dense>
148 typedef const TensorMap<PlainObjectType, Options, MakePointer>& type;
152 struct eval<const TensorMap<PlainObjectType, Options, MakePointer>, Eigen::Dense>
154 typedef const TensorMap<PlainObjectType, Options, MakePointer>& type;
201 struct nested<TensorMap<PlainObjectType, Options, MakePointer> >
203 typedef const TensorMap<PlainObjectType, Options, MakePointer>& type;
207 struct nested<const TensorMap<PlainObjectType, Options, MakePointer> >
209 typedef const TensorMap<PlainObjectType, Options, MakePointer>& type;
|
| TensorMap.h | 15 /** \class TensorMap
27 template<typename PlainObjectType, int Options_, template <class> class MakePointer_> class TensorMap : public TensorBase<TensorMap<PlainObjectType, Options_, MakePointer_> >
30 typedef TensorMap<PlainObjectType, Options_, MakePointer_> Self;
60 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr) : m_data(dataPtr), m_dimensions() {
67 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index firstDimension, IndexTypes... otherDimensions) : m_data(dataPtr), m_dimensions(firstDimension, otherDimensions...) {
73 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index firstDimension) : m_data(dataPtr), m_dimensions(firstDimension) {
78 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2) : m_data(dataPtr), m_dimensions(dim1, dim2) {
82 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2, Index dim3) : m_data(dataPtr), m_dimensions(dim1, dim2, dim3) {
86 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2, Index dim3, Index dim4) : m_data(dataPtr), m_dimen (…)
[all...] |
| TensorSyclLeafCount.h | 46 /// specialisation of the \ref LeafCount struct when the node type is const TensorMap
48 struct LeafCount<const TensorMap<PlainObjectType, Options_, MakePointer_> > {
52 /// specialisation of the \ref LeafCount struct when the node type is TensorMap
54 struct LeafCount<TensorMap<PlainObjectType, Options_, MakePointer_> > :LeafCount<const TensorMap<PlainObjectType, Options_, MakePointer_> >{};
|
| TensorSyclConvertToDeviceExpression.h | 49 /// type is TensorMap
52 struct ConvertToDeviceExpression<CVQual TensorMap<Tensor<Scalar_, NumIndices_, Options_, IndexType_>, Options2_, MakePointer_> > {\
53 typedef CVQual TensorMap<Tensor<Scalar_, NumIndices_, Options_, IndexType_>, Options2_, MakeGlobalPointer> Type;\
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_conv2d_impl.h | 38 const Eigen::TensorMap<Eigen::Tensor<const ScalarType, 4, Eigen::RowMajor>,
42 const Eigen::TensorMap<Eigen::Tensor<const ScalarType, 4, Eigen::RowMajor>,
46 Eigen::TensorMap<Eigen::Tensor<ScalarType, 4, Eigen::RowMajor>,
|
| runtime_matmul.cc | 57 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> A(lhs, lhs_rows,
59 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> B(rhs, rhs_rows,
61 Eigen::TensorMap<Eigen::Tensor<T, 2>, Alignment> C(out, m, n);
|
| runtime_single_threaded_matmul.cc | 51 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> A(lhs, lhs_rows,
53 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> B(rhs, rhs_rows,
55 Eigen::TensorMap<Eigen::Tensor<T, 2>, Alignment> C(out, m, n);
|
| runtime_fft_impl.h | 50 const Eigen::TensorMap<Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor>,
53 Eigen::TensorMap<Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor>,
76 const Eigen::TensorMap<Eigen::Tensor<float, FFTRank + 1, Eigen::RowMajor>,
79 Eigen::TensorMap<Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor>,
115 const Eigen::TensorMap<Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor>,
118 Eigen::TensorMap<Eigen::Tensor<float, FFTRank + 1, Eigen::RowMajor>,
|
| /external/tensorflow/tensorflow/examples/ios/camera/ |
| tensorflow_utils.h | 47 void GetTopN(const Eigen::TensorMap<Eigen::Tensor<float, 1, Eigen::RowMajor>,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| matmul_op.h | 34 typedef Eigen::TensorMap<Eigen::Tensor<T, 2, Eigen::RowMajor>, Eigen::Aligned>
36 typedef Eigen::TensorMap<Eigen::Tensor<const T, 2, Eigen::RowMajor>,
|
| sdca_internal.h | 157 Eigen::TensorMap<Eigen::Tensor<const float, 1, Eigen::RowMajor>> Row()
159 return Eigen::TensorMap<Eigen::Tensor<const float, 1, Eigen::RowMajor>>(
165 Eigen::TensorMap<Eigen::Tensor<const float, 2, Eigen::RowMajor>>
167 return Eigen::TensorMap<Eigen::Tensor<const float, 2, Eigen::RowMajor>>(
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| multithreaded_conv.h | 40 typedef Eigen::TensorMap<
43 typedef Eigen::TensorMap<
48 typedef Eigen::TensorMap<
51 typedef Eigen::TensorMap<
|
