Home
last modified time | relevance | path

Searched refs:DimPair (Results 1 – 19 of 19) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_contraction.cpp17 typedef Tensor<float, 1>::DimensionPair DimPair; typedef
32 Eigen::array<DimPair, 1> dims3 = {{DimPair(0, 0)}}; in test_evals()
52 Eigen::array<DimPair, 1> dims4 = {{DimPair(1, 1)}}; in test_evals()
67 Eigen::array<DimPair, 1> dims6 = {{DimPair(1, 0)}}; in test_evals()
90 Eigen::array<DimPair, 1> dims = {{DimPair(0, 0)}}; in test_scalar()
111 Eigen::array<DimPair, 2> dims = {{DimPair(1, 2), DimPair(2, 3)}}; in test_multidims()
145 Eigen::array<DimPair, 2> dims2({{DimPair(0, 1), DimPair(1, 0)}}); in test_multidims()
165 Eigen::array<DimPair, 2> dims = {{DimPair(0, 0), DimPair(3, 4)}}; in test_holes()
200 Eigen::array<DimPair, 2> dims = {{DimPair(0, 0), DimPair(1, 1)}}; in test_full_redux()
208 dims[0] = DimPair(1, 0); in test_full_redux()
[all …]
Dcxx11_tensor_custom_op.cpp74 typedef Tensor<float, 3>::DimensionPair DimPair; in eval() typedef
75 array<DimPair, 1> dims; in eval()
76 dims[0] = DimPair(1, 0); in eval()
93 typedef Tensor<float, 3>::DimensionPair DimPair; in test_custom_binary_op() typedef
94 array<DimPair, 1> dims; in test_custom_binary_op()
95 dims[0] = DimPair(1, 0); in test_custom_binary_op()
Dcxx11_tensor_thread_pool.cpp77 typedef Tensor<float, 1>::DimensionPair DimPair; in test_multithread_contraction() typedef
78 Eigen::array<DimPair, 2> dims({{DimPair(2, 0), DimPair(3, 1)}}); in test_multithread_contraction()
118 typedef Tensor<float, 1>::DimensionPair DimPair; in test_contraction_corner_cases() typedef
119 Eigen::array<DimPair, 1> dims{{DimPair(0, 0)}}; in test_contraction_corner_cases()
213 typedef Tensor<float, 1>::DimensionPair DimPair; in test_multithread_contraction_agrees_with_singlethread() typedef
214 Eigen::array<DimPair, 1> dims({{DimPair(1, 2)}}); in test_multithread_contraction_agrees_with_singlethread()
252 typedef Tensor<float, 2>::DimensionPair DimPair; in test_full_contraction() typedef
253 Eigen::array<DimPair, 2> dims({{DimPair(0, 0), DimPair(1, 1)}}); in test_full_contraction()
Dcxx11_tensor_contract_cuda.cu24 typedef Tensor<float, 1>::DimensionPair DimPair; typedef
37 Eigen::array<DimPair, 1> dims(DimPair(1, 0)); in test_cuda_contraction()
101 Eigen::array<DimPair, 2> dims(DimPair(0, 0), DimPair(1, 1)); in test_scalar()
Dcxx11_tensor_of_complex.cpp85 typedef Tensor<float, 1>::DimensionPair DimPair; in test_contractions() typedef
86 Eigen::array<DimPair, 2> dims; in test_contractions()
87 dims[0] = DimPair(2, 0); in test_contractions()
88 dims[1] = DimPair(3, 1); in test_contractions()
Dcxx11_tensor_forced_eval.cpp31 typedef Tensor<float, 1>::DimensionPair DimPair; in test_simple() typedef
32 Eigen::array<DimPair, 1> dims; in test_simple()
33 dims[0] = DimPair(1, 0); in test_simple()
Dcxx11_tensor_morphing.cpp56 typedef Tensor<float, 1>::DimensionPair DimPair; in test_reshape_in_expr() typedef
57 array<DimPair, 1> contract_along{{DimPair(1, 0)}}; in test_reshape_in_expr()
142 typedef Tensor<float, 1>::DimensionPair DimPair; in test_slice_in_expr() typedef
143 array<DimPair, 1> contract_along{{DimPair(1, 0)}}; in test_slice_in_expr()
Dcxx11_tensor_scan_cuda.cu23 typedef Tensor<float, 1>::DimensionPair DimPair; typedef
Dcxx11_tensor_of_float16_cuda.cu291 typedef Tensor<float, 2>::DimensionPair DimPair; in test_cuda_contractions() typedef
292 Eigen::array<DimPair, 1> dims(DimPair(1, 0)); in test_cuda_contractions()
Dcxx11_tensor_cuda.cu296 typedef Tensor<float, 1>::DimensionPair DimPair; in test_cuda_contraction() typedef
297 Eigen::array<DimPair, 2> dims; in test_cuda_contraction()
298 dims[0] = DimPair(2, 0); in test_cuda_contraction()
299 dims[1] = DimPair(3, 1); in test_cuda_contraction()
/external/eigen/bench/btl/libs/tensors/
Dtensor_interface.hh77 typedef typename Eigen::Tensor<real_type, 1>::DimensionPair DimPair; in matrix_matrix_product() typedef
78 const Eigen::array<DimPair, 1> dims(DimPair(1, 0)); in matrix_matrix_product()
83 typedef typename Eigen::Tensor<real_type, 1>::DimensionPair DimPair; in matrix_vector_product() typedef
84 const Eigen::array<DimPair, 1> dims(DimPair(1, 0)); in matrix_vector_product()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_matmul.cc54 typedef typename Eigen::Tensor<T, 2>::DimensionPair DimPair; in MatMul() typedef
57 const Eigen::array<DimPair, 1> dims( in MatMul()
58 {DimPair(lhs_contract_dim, rhs_contract_dim)}); in MatMul()
Druntime_matmul.cc61 typedef typename Eigen::Tensor<T, 2>::DimensionPair DimPair; in MatMul() typedef
64 const Eigen::array<DimPair, 1> dims( in MatMul()
65 {DimPair(lhs_contract_dim, rhs_contract_dim)}); in MatMul()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.h42 typename DimPair>
44 const DimPair& dim_pair) { in MatMul()
Dlrn_op.cc113 Eigen::array<DimPair, 1> dims = {{DimPair(1, 0)}}; in launch()
128 typedef typename Eigen::Tensor<T, 1, Eigen::RowMajor>::DimensionPair DimPair; typedef
Deigen_spatial_convolutions_test.cc1285 typedef Tensor<float, 1>::DimensionPair DimPair; in TEST() typedef
1323 array<DimPair, 2> contract_dims; in TEST()
1324 contract_dims[0] = DimPair(0, 0); in TEST()
1325 contract_dims[1] = DimPair(2, 1); in TEST()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_lrn_op.cc248 Eigen::array<DimPair, 1> dims = {{DimPair(1, 0)}}; in MklDefaultToEigen()
303 typedef typename Eigen::Tensor<T, 1, Eigen::RowMajor>::DimensionPair DimPair; typedef in tensorflow::MklLRNOp
650 typedef typename Eigen::Tensor<T, 1, Eigen::RowMajor>::DimensionPair DimPair; typedef in tensorflow::MklLRNGradOp
/external/eigen/bench/tensors/
Dtensor_benchmarks.h404 typedef typename Tensor<T, 2>::DimensionPair DimPair; in contraction() typedef
405 Eigen::array<DimPair, 1> dims; in contraction()
406 dims[0] = DimPair(1, 0); in contraction()
/external/ruy/ruy/
Dtest.h1063 using DimPair =
1065 Eigen::array<DimPair, 1> contract_dims(
1066 {DimPair((LhsOrder == Order::kColMajor) ? 1 : 0,