Home
last modified time | relevance | path

Searched refs:RowMajor (Results 1 – 25 of 343) sorted by relevance

12345678910>>...14

/external/gemmlowp/meta/
Dquantized_mul_kernels_arm_64.h28 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 1,
31 RowMajor>& params, in Multiply()
101 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 2,
104 RowMajor>& params, in Multiply()
177 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 3,
180 RowMajor>& params, in Multiply()
259 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 4,
262 RowMajor>& params, in Multiply()
343 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 5,
346 RowMajor>& params, in Multiply()
[all …]
Dquantized_mul_kernels_arm_32.h28 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 1,
31 RowMajor>& params, in Multiply()
102 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 2,
105 RowMajor>& params, in Multiply()
180 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 3,
183 RowMajor>& params, in Multiply()
264 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 4,
267 RowMajor>& params, in Multiply()
352 MulKernel<uint8_t, uint8_t, QuantizedStaticPreprocessed, RowMajor, 1, 5,
355 RowMajor>& params, in Multiply()
[all …]
Dstreams.h25 struct RowMajor { struct
48 class StreamUtil<InType, RowMajor> { argument
50 static const InType* Offset(const RowMajor& params, const InType* source, in Offset()
57 static InType* Offset(const RowMajor& params, InType* source, in Offset()
64 static int Scratch(const RowMajor& params, int lanes_count, int pack_size) { in Scratch()
120 class Stream<InType, lanes_count, pack_size, leftovers, RowMajor> {
122 static void Pack(const InType* in, const RowMajor& params, InType* out) { in Pack()
137 static int UnpackedAdvance(const RowMajor& params) { in UnpackedAdvance()
141 static int PackedAdvance(const RowMajor& params) { in PackedAdvance()
145 static int UnpackedStride(const RowMajor& params) { in UnpackedStride()
[all …]
Dquantized_mul_kernels.h36 class MulKernel<InType, OutType, QuantizedStaticPreprocessed, RowMajor, m, n,
39 typedef FusedKernelParams<QuantizedStaticPreprocessed, RowMajor> FusedKernel;
84 class MulKernel<InType, OutType, QuantizedStaticPreprocessedAsInt32, RowMajor,
87 typedef FusedKernelParams<QuantizedStaticPreprocessedAsInt32, RowMajor>
129 class MulKernel<InType, OutType, QuantizedStaticPreprocessedAsFloat, RowMajor,
132 typedef FusedKernelParams<QuantizedStaticPreprocessedAsFloat, RowMajor>
/external/tensorflow/tensorflow/core/framework/
Dtensor_types.h27 typedef Eigen::TensorMap<Eigen::Tensor<T, NDIMS, Eigen::RowMajor, IndexType>,
31 Eigen::Tensor<const T, NDIMS, Eigen::RowMajor, IndexType>, Eigen::Aligned>
35 typedef Eigen::TensorMap<Eigen::Tensor<T, NDIMS, Eigen::RowMajor, IndexType> >
38 Eigen::Tensor<const T, NDIMS, Eigen::RowMajor, IndexType> >
41 typedef Eigen::TensorMap<Eigen::Tensor<T, NDIMS, Eigen::RowMajor, int>,
47 Eigen::TensorFixedSize<T, Eigen::Sizes<>, Eigen::RowMajor, IndexType>,
51 Eigen::RowMajor, IndexType>,
57 Eigen::TensorFixedSize<T, Eigen::Sizes<>, Eigen::RowMajor, IndexType> >
60 Eigen::RowMajor, IndexType> >
64 typedef Eigen::TensorMap<Eigen::Tensor<T, 1, Eigen::RowMajor, IndexType>,
[all …]
/external/eigen/blas/
Dlevel2_impl.h18 typedef internal::const_blas_data_mapper<Scalar,Index,RowMajor> RhsMapper; in run()
34 (general_matrix_vector_product_wrapper<int,Scalar,RowMajor,false,false>::run), in EIGEN_BLAS_FUNC()
36 (general_matrix_vector_product_wrapper<int,Scalar,RowMajor,Conj ,false>::run), in EIGEN_BLAS_FUNC()
93 …(internal::triangular_solve_vector<Scalar,Scalar,int,OnTheLeft, Lower|0, false,RowMajor>::ru… in EIGEN_BLAS_FUNC()
95 …(internal::triangular_solve_vector<Scalar,Scalar,int,OnTheLeft, Lower|0, Conj, RowMajor>::ru… in EIGEN_BLAS_FUNC()
100 …(internal::triangular_solve_vector<Scalar,Scalar,int,OnTheLeft, Upper|0, false,RowMajor>::ru… in EIGEN_BLAS_FUNC()
102 …(internal::triangular_solve_vector<Scalar,Scalar,int,OnTheLeft, Upper|0, Conj, RowMajor>::ru… in EIGEN_BLAS_FUNC()
107 …(internal::triangular_solve_vector<Scalar,Scalar,int,OnTheLeft, Lower|UnitDiag,false,RowMajor>::ru… in EIGEN_BLAS_FUNC()
109 …(internal::triangular_solve_vector<Scalar,Scalar,int,OnTheLeft, Lower|UnitDiag,Conj, RowMajor>::ru… in EIGEN_BLAS_FUNC()
114 …(internal::triangular_solve_vector<Scalar,Scalar,int,OnTheLeft, Upper|UnitDiag,false,RowMajor>::ru… in EIGEN_BLAS_FUNC()
[all …]
Dlevel3_impl.h21 …(internal::general_matrix_matrix_product<DenseIndex,Scalar,RowMajor,false,Scalar,ColMajor,false,Co… in EIGEN_BLAS_FUNC()
23 …(internal::general_matrix_matrix_product<DenseIndex,Scalar,RowMajor,Conj, Scalar,ColMajor,false,Co… in EIGEN_BLAS_FUNC()
26 …(internal::general_matrix_matrix_product<DenseIndex,Scalar,ColMajor,false,Scalar,RowMajor,false,Co… in EIGEN_BLAS_FUNC()
28 …(internal::general_matrix_matrix_product<DenseIndex,Scalar,RowMajor,false,Scalar,RowMajor,false,Co… in EIGEN_BLAS_FUNC()
30 …(internal::general_matrix_matrix_product<DenseIndex,Scalar,RowMajor,Conj, Scalar,RowMajor,false,Co… in EIGEN_BLAS_FUNC()
33 …(internal::general_matrix_matrix_product<DenseIndex,Scalar,ColMajor,false,Scalar,RowMajor,Conj, Co… in EIGEN_BLAS_FUNC()
35 …(internal::general_matrix_matrix_product<DenseIndex,Scalar,RowMajor,false,Scalar,RowMajor,Conj, Co… in EIGEN_BLAS_FUNC()
37 …(internal::general_matrix_matrix_product<DenseIndex,Scalar,RowMajor,Conj, Scalar,RowMajor,Conj, Co… in EIGEN_BLAS_FUNC()
87 …angular_solve_matrix<Scalar,DenseIndex,OnTheLeft, Lower|0, false,RowMajor,ColMajor>::run), in EIGEN_BLAS_FUNC()
89 …ngular_solve_matrix<Scalar,DenseIndex,OnTheLeft, Lower|0, Conj, RowMajor,ColMajor>::run),\ in EIGEN_BLAS_FUNC()
[all …]
/external/eigen/Eigen/src/SparseCore/
DSparseSparseProductWithPruning.h100 struct sparse_sparse_product_with_pruning_selector<Lhs,Rhs,ResultType,ColMajor,ColMajor,RowMajor>
114 struct sparse_sparse_product_with_pruning_selector<Lhs,Rhs,ResultType,RowMajor,RowMajor,RowMajor>
127 struct sparse_sparse_product_with_pruning_selector<Lhs,Rhs,ResultType,RowMajor,RowMajor,ColMajor>
147 struct sparse_sparse_product_with_pruning_selector<Lhs,Rhs,ResultType,ColMajor,RowMajor,RowMajor>
152 …typedef SparseMatrix<typename ResultType::Scalar,RowMajor,typename Lhs::StorageIndex> RowMajorMatr…
154 …sparse_sparse_product_with_pruning_selector<RowMajorMatrixLhs,Rhs,ResultType,RowMajor,RowMajor>(ro…
159 struct sparse_sparse_product_with_pruning_selector<Lhs,Rhs,ResultType,RowMajor,ColMajor,RowMajor>
164 …typedef SparseMatrix<typename ResultType::Scalar,RowMajor,typename Lhs::StorageIndex> RowMajorMatr…
166 …product_with_pruning_selector<Lhs,RowMajorMatrixRhs,ResultType,RowMajor,RowMajor,RowMajor>(lhs,row…
171 struct sparse_sparse_product_with_pruning_selector<Lhs,Rhs,ResultType,ColMajor,RowMajor,ColMajor>
[all …]
DConservativeSparseSparseProduct.h126 int LhsStorageOrder = (traits<Lhs>::Flags&RowMajorBit) ? RowMajor : ColMajor,
127 int RhsStorageOrder = (traits<Rhs>::Flags&RowMajorBit) ? RowMajor : ColMajor,
128 int ResStorageOrder = (traits<ResultType>::Flags&RowMajorBit) ? RowMajor : ColMajor>
139 …typedef SparseMatrix<typename ResultType::Scalar,RowMajor,typename ResultType::StorageIndex> RowMa…
165 struct conservative_sparse_sparse_product_selector<Lhs,Rhs,ResultType,RowMajor,ColMajor,ColMajor>
169 …typedef SparseMatrix<typename ResultType::Scalar,RowMajor,typename ResultType::StorageIndex> RowMa…
178 struct conservative_sparse_sparse_product_selector<Lhs,Rhs,ResultType,ColMajor,RowMajor,ColMajor>
182 …typedef SparseMatrix<typename ResultType::Scalar,RowMajor,typename ResultType::StorageIndex> RowMa…
191 struct conservative_sparse_sparse_product_selector<Lhs,Rhs,ResultType,RowMajor,RowMajor,ColMajor>
195 …typedef SparseMatrix<typename ResultType::Scalar,RowMajor,typename ResultType::StorageIndex> RowMa…
[all …]
/external/eigen/test/
Dproduct_trmm.cpp27 typedef Matrix<Scalar,OtherCols,Dynamic,OtherCols==1?RowMajor:OtherOrder> OnTheLeft; in trmm()
30 typedef Matrix<Scalar,OtherCols,Dynamic,OtherCols==1?RowMajor:ResOrder> ResSX; in trmm()
85 EIGEN_CAT(CALL_SUBTEST_,NB)((trmm<SCALAR, MODE, ColMajor,ColMajor,RowMajor>())); \
86 EIGEN_CAT(CALL_SUBTEST_,NB)((trmm<SCALAR, MODE, ColMajor,RowMajor,ColMajor>())); \
87 EIGEN_CAT(CALL_SUBTEST_,NB)((trmm<SCALAR, MODE, ColMajor,RowMajor,RowMajor>())); \
88 EIGEN_CAT(CALL_SUBTEST_,NB)((trmm<SCALAR, MODE, RowMajor,ColMajor,ColMajor>())); \
89 EIGEN_CAT(CALL_SUBTEST_,NB)((trmm<SCALAR, MODE, RowMajor,ColMajor,RowMajor>())); \
90 EIGEN_CAT(CALL_SUBTEST_,NB)((trmm<SCALAR, MODE, RowMajor,RowMajor,ColMajor>())); \
91 EIGEN_CAT(CALL_SUBTEST_,NB)((trmm<SCALAR, MODE, RowMajor,RowMajor,RowMajor>())); \
94 EIGEN_CAT(CALL_SUBTEST_1,NB)((trmv<SCALAR, MODE, RowMajor>()));
Dproduct_small.cpp45 || (Depth==1&&Rows !=1&&OA==RowMajor)
46 || (Cols ==1&&Depth!=1&&OB==RowMajor)
49 || (Cols ==1&&Rows !=1&&OC==RowMajor)),void>::type
61 || (Depth==1&&Rows !=1&&OA==RowMajor)
62 || (Cols ==1&&Depth!=1&&OB==RowMajor)
65 || (Cols ==1&&Rows !=1&&OC==RowMajor)),void>::type
74 CALL_SUBTEST(( test_lazy_single<T,Rows,Cols,Depth,RowMajor,ColMajor,ColMajor>(rows,cols,depth) )); in test_lazy_all_layout()
75 CALL_SUBTEST(( test_lazy_single<T,Rows,Cols,Depth,ColMajor,RowMajor,ColMajor>(rows,cols,depth) )); in test_lazy_all_layout()
76 CALL_SUBTEST(( test_lazy_single<T,Rows,Cols,Depth,RowMajor,RowMajor,ColMajor>(rows,cols,depth) )); in test_lazy_all_layout()
77 CALL_SUBTEST(( test_lazy_single<T,Rows,Cols,Depth,ColMajor,ColMajor,RowMajor>(rows,cols,depth) )); in test_lazy_all_layout()
[all …]
Dpardiso_support.cpp10 PardisoLLT < SparseMatrix<T, RowMajor>, Lower> pardiso_llt_lower; in test_pardiso_T()
11 PardisoLLT < SparseMatrix<T, RowMajor>, Upper> pardiso_llt_upper; in test_pardiso_T()
12 PardisoLDLT < SparseMatrix<T, RowMajor>, Lower> pardiso_ldlt_lower; in test_pardiso_T()
13 PardisoLDLT < SparseMatrix<T, RowMajor>, Upper> pardiso_ldlt_upper; in test_pardiso_T()
14 PardisoLU < SparseMatrix<T, RowMajor> > pardiso_lu; in test_pardiso_T()
/external/eigen/unsupported/test/
Dcxx11_tensor_fft.cpp232 test_fft_2D_golden<RowMajor>(); in test_cxx11_tensor_fft()
254 test_fft_real_input_energy<RowMajor, float, true, Eigen::BothParts, FFT_FORWARD, 1>(); in test_cxx11_tensor_fft()
255 test_fft_real_input_energy<RowMajor, double, true, Eigen::BothParts, FFT_FORWARD, 1>(); in test_cxx11_tensor_fft()
256 test_fft_real_input_energy<RowMajor, float, false, Eigen::BothParts, FFT_FORWARD, 1>(); in test_cxx11_tensor_fft()
257 test_fft_real_input_energy<RowMajor, double, false, Eigen::BothParts, FFT_FORWARD, 1>(); in test_cxx11_tensor_fft()
259 test_fft_real_input_energy<RowMajor, float, true, Eigen::BothParts, FFT_FORWARD, 2>(); in test_cxx11_tensor_fft()
260 test_fft_real_input_energy<RowMajor, double, true, Eigen::BothParts, FFT_FORWARD, 2>(); in test_cxx11_tensor_fft()
261 test_fft_real_input_energy<RowMajor, float, false, Eigen::BothParts, FFT_FORWARD, 2>(); in test_cxx11_tensor_fft()
262 test_fft_real_input_energy<RowMajor, double, false, Eigen::BothParts, FFT_FORWARD, 2>(); in test_cxx11_tensor_fft()
264 test_fft_real_input_energy<RowMajor, float, true, Eigen::BothParts, FFT_FORWARD, 3>(); in test_cxx11_tensor_fft()
[all …]
Dcxx11_tensor_expr.cpp15 using Eigen::RowMajor;
20 Tensor<float, 1, RowMajor> vec2(6); in test_1d()
33 TensorMap<Tensor<float, 1, RowMajor>> vec4(data4, 6); in test_1d()
36 TensorMap<Tensor<float, 1, RowMajor>> vec5(data5, 6); in test_1d()
74 TensorMap<Tensor<float, 2, RowMajor>> mat2(data2, 2, 3); in test_2d()
91 Tensor<float, 2, RowMajor> mat4(2,3); in test_2d()
113 Tensor<float, 3, RowMajor> mat2(2,3,7); in test_3d()
128 Tensor<float, 3, RowMajor> mat4(2,3,7); in test_3d()
132 Tensor<float, 3, RowMajor> mat6(2,3,7); in test_3d()
136 Tensor<float, 3, RowMajor> mat8(2,3,7); in test_3d()
[all …]
Dcxx11_tensor_map.cpp15 using Eigen::RowMajor;
20 Tensor<int, 0, RowMajor> scalar2; in test_0d()
23 TensorMap<Tensor<const int, 0, RowMajor> > scalar4(scalar2.data()); in test_0d()
38 Tensor<int, 1, RowMajor> vec2(6); in test_1d()
41 TensorMap<Tensor<const int, 1, RowMajor> > vec4(vec2.data(), 6); in test_1d()
72 Tensor<int, 2, RowMajor> mat2(2,3); in test_2d()
89 TensorMap<Tensor<const int, 2, RowMajor> > mat4(mat2.data(), 2, 3); in test_2d()
119 Tensor<int, 3, RowMajor> mat2(2,3,7); in test_3d()
133 TensorMap<Tensor<const int, 3, RowMajor> > mat4(mat2.data(), 2, 3, 7); in test_3d()
163 Tensor<int, 3, RowMajor> mat2(2,3,7); in test_from_tensor()
[all …]
Dcxx11_tensor_fixed_size.cpp15 using Eigen::RowMajor;
21 TensorFixedSize<float, Sizes<>, RowMajor> scalar2; in test_0d()
38 TensorFixedSize<float, Sizes<>, RowMajor> scalar4 = scalar2.sqrt(); in test_0d()
50 TensorFixedSize<float, Sizes<6>, RowMajor> vec2; in test_1d()
76 TensorFixedSize<float, Sizes<6>, RowMajor> vec4 = vec2.sqrt(); in test_1d()
109 TensorFixedSize<float, Sizes<6>, RowMajor> vec2; in test_tensor_map()
135 TensorMap<TensorFixedSize<float, Sizes<2, 3>, RowMajor> > mat2(data2,2,3); in test_2d()
157 TensorFixedSize<float, Sizes<2, 3>, RowMajor> mat4; in test_2d()
183 TensorFixedSize<float, Sizes<2, 3, 7>, RowMajor> mat2; in test_3d()
204 TensorFixedSize<float, Sizes<2, 3, 7>, RowMajor> mat4; in test_3d()
Dcxx11_tensor_contract_cuda.cu152 test_cuda_contraction<RowMajor>(k, 128, 128); in test_cuda_contraction_m()
160 test_cuda_contraction<RowMajor>(128, k, 128); in test_cuda_contraction_k()
168 test_cuda_contraction<RowMajor>(128, 128, k); in test_cuda_contraction_n()
200 CALL_SUBTEST_1(test_cuda_contraction<RowMajor>(128, 128, 128)); in test_cxx11_tensor_cuda()
203 CALL_SUBTEST_1(test_scalar<RowMajor>(128, 128, 128)); in test_cxx11_tensor_cuda()
206 CALL_SUBTEST_3(test_cuda_contraction_m<RowMajor>()); in test_cxx11_tensor_cuda()
209 CALL_SUBTEST_5(test_cuda_contraction_k<RowMajor>()); in test_cxx11_tensor_cuda()
212 CALL_SUBTEST_7(test_cuda_contraction_n<RowMajor>()); in test_cxx11_tensor_cuda()
215 CALL_SUBTEST_9(test_cuda_contraction_sizes<RowMajor>()); in test_cxx11_tensor_cuda()
Dcxx11_tensor_io.cpp125 CALL_SUBTEST(test_output_0d<RowMajor>()); in test_cxx11_tensor_io()
127 CALL_SUBTEST(test_output_1d<RowMajor>()); in test_cxx11_tensor_io()
129 CALL_SUBTEST(test_output_2d<RowMajor>()); in test_cxx11_tensor_io()
131 CALL_SUBTEST(test_output_expr<RowMajor>()); in test_cxx11_tensor_io()
133 CALL_SUBTEST(test_output_string<RowMajor>()); in test_cxx11_tensor_io()
135 CALL_SUBTEST(test_output_const<RowMajor>()); in test_cxx11_tensor_io()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDimensions.h40 template<typename Index, std::size_t NumIndices, std::size_t n, bool RowMajor>
47 return array_get<RowMajor ? n - 1 : (NumIndices - n)>(indices) + in run()
48 dget<RowMajor ? n - 1 : (NumIndices - n), Dimensions>::value * in run()
49 …fixed_size_tensor_index_linearization_helper<Index, NumIndices, n - 1, RowMajor>::run(indices, dim… in run()
53 template<typename Index, std::size_t NumIndices, bool RowMajor>
54 struct fixed_size_tensor_index_linearization_helper<Index, NumIndices, 0, RowMajor>
234 template<typename Index, std::size_t NumIndices, std::size_t n, bool RowMajor>
240 return array_get<RowMajor ? n : (NumIndices - n - 1)>(indices) +
241 array_get<RowMajor ? n : (NumIndices - n - 1)>(dimensions) *
242 … tensor_index_linearization_helper<Index, NumIndices, n - 1, RowMajor>::run(indices, dimensions);
[all …]
DTensorMap.h125 if (PlainObjectType::Options&RowMajor) { in operator()
153 if (PlainObjectType::Options&RowMajor) { in operator()
165 if (PlainObjectType::Options&RowMajor) { in operator()
176 if (PlainObjectType::Options&RowMajor) { in operator()
187 if (PlainObjectType::Options&RowMajor) { in operator()
198 if (PlainObjectType::Options&RowMajor) { in operator()
212 if (PlainObjectType::Options&RowMajor) { in operator()
241 if (PlainObjectType::Options&RowMajor) { in operator()
253 if (PlainObjectType::Options&RowMajor) { in operator()
264 if (PlainObjectType::Options&RowMajor) { in operator()
[all …]
/external/eigen/Eigen/src/Core/products/
DGeneralMatrixVector_BLAS.h54 …Major>,ColMajor,ConjugateLhs,Scalar,const_blas_data_mapper<Scalar,Index,RowMajor>,ConjugateRhs,Spe…
58 const const_blas_data_mapper<Scalar,Index,RowMajor> &rhs, \
62 …Major>,ColMajor,ConjugateLhs,Scalar,const_blas_data_mapper<Scalar,Index,RowMajor>,ConjugateRhs,Bui…
71 …ix_vector_product<Index,Scalar,const_blas_data_mapper<Scalar,Index,RowMajor>,RowMajor,ConjugateLhs…
74 const const_blas_data_mapper<Scalar,Index,RowMajor> &lhs, \
78 … general_matrix_vector_product_gemv<Index,Scalar,RowMajor,ConjugateLhs,Scalar,ConjugateRhs>::run( \
105 if (LhsStorageOrder==RowMajor) { \
DGeneralMatrixMatrix.h24 …oduct<Index,LhsScalar,LhsStorageOrder,ConjugateLhs,RhsScalar,RhsStorageOrder,ConjugateRhs,RowMajor>
40 RhsScalar, RhsStorageOrder==RowMajor ? ColMajor : RowMajor, ConjugateRhs,
41 LhsScalar, LhsStorageOrder==RowMajor ? ColMajor : RowMajor, ConjugateLhs,
276 typename conditional<StorageOrder==RowMajor,_RhsScalar,_LhsScalar>::type,
277 typename conditional<StorageOrder==RowMajor,_LhsScalar,_RhsScalar>::type>
280 Transpose = StorageOrder==RowMajor,
327 typename conditional<StorageOrder==RowMajor,_RhsScalar,_LhsScalar>::type,
328 typename conditional<StorageOrder==RowMajor,_LhsScalar,_RhsScalar>::type>
331 Transpose = StorageOrder==RowMajor
469 …typedef internal::gemm_blocking_space<(Dest::Flags&RowMajorBit) ? RowMajor : ColMajor,LhsScalar,Rh…
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc89 Tensor<float, 3, RowMajor> input_backward(input_cols, input_rows, in TEST()
91 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, in TEST()
93 Tensor<float, 3, RowMajor> output_backward(output_cols, output_rows, in TEST()
207 Tensor<float, 4, RowMajor> input_backward(input_cols, input_rows, in TEST()
209 Tensor<float, 5, RowMajor> kernel(patch_cols, patch_rows, patch_planes, in TEST()
211 Tensor<float, 4, RowMajor> output_backward(output_cols, output_rows, in TEST()
315 Tensor<float, 3, RowMajor> input_backward(input_cols, input_rows, in TEST()
317 Tensor<float, 4, RowMajor> kernel(patch_cols, patch_rows, input_depth, in TEST()
319 Tensor<float, 3, RowMajor> output_backward(output_cols, output_rows, in TEST()
440 Tensor<float, 4, RowMajor> input_backward(input_cols, input_rows, in TEST()
[all …]
Dsdca_internal.h96 Eigen::Tensor<float, 1, Eigen::RowMajor> EigenShrinkVector( in EigenShrinkVector()
97 const Eigen::Tensor<float, 1, Eigen::RowMajor> weights) const { in EigenShrinkVector()
104 Eigen::Tensor<float, 2, Eigen::RowMajor> EigenShrinkMatrix( in EigenShrinkMatrix()
105 const Eigen::Tensor<float, 2, Eigen::RowMajor> weights) const { in EigenShrinkMatrix()
157 Eigen::TensorMap<Eigen::Tensor<const float, 1, Eigen::RowMajor>> Row() in Row()
159 return Eigen::TensorMap<Eigen::Tensor<const float, 1, Eigen::RowMajor>>( in Row()
165 Eigen::TensorMap<Eigen::Tensor<const float, 2, Eigen::RowMajor>>
167 return Eigen::TensorMap<Eigen::Tensor<const float, 2, Eigen::RowMajor>>( in RowAsMatrix()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h50 const Eigen::TensorMap<Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor>, in EigenFftC2C()
53 Eigen::TensorMap<Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor>, in EigenFftC2C()
76 const Eigen::TensorMap<Eigen::Tensor<float, FFTRank + 1, Eigen::RowMajor>, in EigenFftR2C()
79 Eigen::TensorMap<Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor>, in EigenFftR2C()
87 Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor> full_fft(in_dims); in EigenFftR2C()
115 const Eigen::TensorMap<Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor>, in EigenFftC2R()
118 Eigen::TensorMap<Eigen::Tensor<float, FFTRank + 1, Eigen::RowMajor>, in EigenFftC2R()
126 Eigen::Tensor<complex64, FFTRank + 1, Eigen::RowMajor> full_fft(out_dims); in EigenFftC2R()

12345678910>>...14