Home
last modified time | relevance | path

Searched defs:dims (Results 1 – 18 of 18) sorted by relevance

/applications/standard/app_samples/code/AI/MindSporeLiteCDemoASR/entry/src/main/cpp/third_party/librosa/eigen3/unsupported/test/
Dcxx11_tensor_contraction.cpp90 Eigen::array<DimPair, 1> dims = {{DimPair(0, 0)}}; in test_scalar() local
111 Eigen::array<DimPair, 2> dims = {{DimPair(1, 2), DimPair(2, 3)}}; in test_multidims() local
165 Eigen::array<DimPair, 2> dims = {{DimPair(0, 0), DimPair(3, 4)}}; in test_holes() local
200 Eigen::array<DimPair, 2> dims = {{DimPair(0, 0), DimPair(1, 1)}}; in test_full_redux() local
230 Eigen::array<DimPair, 1> dims = {{DimPair(1, 0)}}; in test_contraction_of_contraction() local
261 Eigen::array<DimPair, 1> dims = {{DimPair(1, 0)}}; in test_expr() local
281 Eigen::array<DimPair, 2> dims = {{DimPair(2, 0), DimPair(0, 2)}}; in test_out_of_order_contraction() local
372 Eigen::array<DimPair, 2> dims = {{DimPair(2, 0), DimPair(3, 1)}}; in test_large_contraction() local
400 Eigen::array<DimPair, 1> dims{{DimPair(1, 0)}}; in test_matrix_vector() local
452 Eigen::array<DimPair, 2> dims = {{DimPair(2, 0), DimPair(3, 1)}}; in test_small_blocking_factors() local
[all …]
Dcxx11_tensor_convolution.cpp54 Eigen::array<ptrdiff_t, 2> dims; in test_expr() local
80 Eigen::array<ptrdiff_t, 1> dims; in test_modes() local
122 Eigen::array<ptrdiff_t, 1> dims; in test_strides() local
Dcxx11_tensor_argmax.cpp65 DimensionList<DenseIndex, 4> dims; in test_argmax_tuple_reducer() local
97 DimensionList<DenseIndex, 4> dims; in test_argmin_tuple_reducer() local
164 std::vector<int> dims {2, 3, 5, 7}; in test_argmax_dim() local
222 std::vector<int> dims {2, 3, 5, 7}; in test_argmin_dim() local
Dcxx11_tensor_thread_pool.cpp78 Eigen::array<DimPair, 2> dims({{DimPair(2, 0), DimPair(3, 1)}}); in test_multithread_contraction() local
119 Eigen::array<DimPair, 1> dims{{DimPair(0, 0)}}; in test_contraction_corner_cases() local
214 Eigen::array<DimPair, 1> dims({{DimPair(1, 2)}}); in test_multithread_contraction_agrees_with_singlethread() local
253 Eigen::array<DimPair, 2> dims({{DimPair(0, 0), DimPair(1, 1)}}); in test_full_contraction() local
Dcxx11_tensor_custom_op.cpp75 array<DimPair, 1> dims; in eval() local
94 array<DimPair, 1> dims; in test_custom_binary_op() local
Dcxx11_tensor_contract_cuda.cu34 Eigen::array<DimPair, 1> dims(DimPair(1, 0)); in test_cuda_contraction() local
98 Eigen::array<DimPair, 2> dims(DimPair(0, 0), DimPair(1, 1)); in test_scalar() local
Dcxx11_tensor_forced_eval.cpp32 Eigen::array<DimPair, 1> dims; in test_simple() local
Dcxx11_tensor_argmax_cuda.cu69 std::vector<int> dims; in test_cuda_argmax_dim() local
158 std::vector<int> dims; in test_cuda_argmin_dim() local
Dcxx11_tensor_device.cu133 Eigen::array<std::pair<int, int>, 2> dims; in test_contraction() local
152 Eigen::array<int, 1> dims(1); in test_1d_convolution() local
162 Eigen::array<int, 2> dims(1,2); in test_2d_convolution() local
172 Eigen::array<int, 3> dims(0,1,2); in test_3d_convolution() local
Dcxx11_tensor_simple.cpp198 array<Eigen::DenseIndex, 3> dims; in test_3d() local
Dcxx11_tensor_of_complex.cpp86 Eigen::array<DimPair, 2> dims; in test_contractions() local
Dcxx11_tensor_cuda.cu294 Eigen::array<DimPair, 2> dims; in test_cuda_contraction() local
345 Eigen::array<Eigen::DenseIndex, 1> dims(1); in test_cuda_convolution_1d() local
398 Eigen::array<Eigen::DenseIndex, 1> dims(0); in test_cuda_convolution_inner_dim_col_major_1d() local
451 Eigen::array<Eigen::DenseIndex, 1> dims(3); in test_cuda_convolution_inner_dim_row_major_1d() local
505 Eigen::array<Eigen::DenseIndex, 2> dims(1,2); in test_cuda_convolution_2d() local
569 Eigen::array<Eigen::DenseIndex, 3> dims(1,2,3); in test_cuda_convolution_3d() local
Dcxx11_tensor_of_float16_cuda.cu289 Eigen::array<DimPair, 1> dims(DimPair(1, 0)); in test_cuda_contractions() local
/applications/standard/app_samples/code/AI/MindSporeLiteCDemoASR/entry/src/main/cpp/third_party/librosa/eigen3/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h485 contract(const OtherDerived& other, const Dimensions& dims) const { in contract()
492 convolve(const KernelDerived& kernel, const Dimensions& dims) const { in convolve()
528 sum(const Dims& dims) const { in sum()
540 mean(const Dims& dims) const { in mean()
552 prod(const Dims& dims) const { in prod()
564 maximum(const Dims& dims) const { in maximum()
576 minimum(const Dims& dims) const { in minimum()
588 all(const Dims& dims) const { in all()
601 any(const Dims& dims) const { in any()
666 reduce(const Dims& dims, const Reducer& reducer) const { in reduce()
DTensor.h451 array<Index, NumIndices> dims; in resize() local
480 array<Index, NumIndices> dims; in resize() local
490 array<Index, NumIndices> dims; in resize() local
DTensorRef.h294 const Dimensions& dims = this->dimensions(); in coeff() local
312 const Dimensions& dims = this->dimensions(); in coeffRef() local
/applications/standard/app_samples/code/AI/MindSporeLiteCDemoASR/entry/src/main/cpp/third_party/librosa/eigen3/bench/btl/libs/tensors/
Dtensor_interface.hh78 const Eigen::array<DimPair, 1> dims(DimPair(1, 0)); in matrix_matrix_product() local
84 const Eigen::array<DimPair, 1> dims(DimPair(1, 0)); in matrix_vector_product() local
/applications/standard/app_samples/code/AI/MindSporeLiteCDemoASR/entry/src/main/cpp/third_party/librosa/eigen3/bench/tensors/
Dtensor_benchmarks.h405 Eigen::array<DimPair, 1> dims; in contraction() local
430 Eigen::array<TensorIndex, 2> dims; in convolution() local