Home
last modified time | relevance | path

Searched defs:DimPair (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_thread_pool.cpp120 typedef Tensor<float, 1>::DimensionPair DimPair; in test_multithread_contraction() typedef
161 typedef Tensor<float, 1>::DimensionPair DimPair; in test_contraction_corner_cases() typedef
256 typedef Tensor<float, 1>::DimensionPair DimPair; in test_multithread_contraction_agrees_with_singlethread() typedef
295 typedef Tensor<float, 1>::DimensionPair DimPair; in test_multithread_contraction_with_output_kernel() typedef
353 typedef Tensor<float, 1>::DimensionPair DimPair; in test_async_multithread_contraction_agrees_with_singlethread() typedef
383 typedef Tensor<float, 1>::DimensionPair DimPair; in test_sharded_by_inner_dim_contraction() typedef
423 typedef Tensor<float, 1>::DimensionPair DimPair; in test_sharded_by_inner_dim_contraction_with_output_kernel() typedef
463 typedef Tensor<float, 1>::DimensionPair DimPair; in test_async_sharded_by_inner_dim_contraction() typedef
507 typedef Tensor<float, 1>::DimensionPair DimPair; in test_async_sharded_by_inner_dim_contraction_with_output_kernel() typedef
562 typedef Tensor<float, 2>::DimensionPair DimPair; in test_full_contraction() typedef
H A Dcxx11_tensor_contract_sycl.cpp39 DimPair; in test_sycl_contraction() typedef
157 DimPair; in test_no_out_of_bounds() typedef
265 DimPair; in test_scalar() typedef
324 DimPair; in contraction_batch() typedef
397 DimPair; in contraction_rhs_transposed() typedef
466 DimPair; in contraction_lhs_transposed() typedef
532 DimPair; in contraction_both_transposed() typedef
H A Dcxx11_tensor_custom_op.cpp74 typedef Tensor<float, 3>::DimensionPair DimPair; in eval() typedef
93 typedef Tensor<float, 3>::DimensionPair DimPair; in test_custom_binary_op() typedef
H A Dcxx11_tensor_custom_op_sycl.cpp101 typedef typename TensorType::DimensionPair DimPair; in eval() typedef
141 typedef typename Eigen::Tensor<DataType, 3, DataLayout, IndexType>::DimensionPair DimPair; in test_custom_binary_op_sycl() typedef
H A Dcxx11_tensor_forced_eval.cpp31 typedef Tensor<float, 1>::DimensionPair DimPair; in test_simple() typedef
H A Dcxx11_tensor_scan_gpu.cu22 typedef Tensor<float, 1>::DimensionPair DimPair; typedef
H A Dcxx11_tensor_morphing.cpp78 typedef Tensor<float, 1>::DimensionPair DimPair; in test_reshape_in_expr() typedef
164 typedef typename Tensor<T, 1>::DimensionPair DimPair; in test_slice_in_expr() typedef
H A Dcxx11_tensor_of_complex.cpp85 typedef Tensor<float, 1>::DimensionPair DimPair; in test_contractions() typedef
H A Dcxx11_tensor_scan_sycl.cpp23 typedef Tensor<float, 1>::DimensionPair DimPair; typedef
H A Dcxx11_tensor_contract_gpu.cu23 typedef Tensor<float, 1>::DimensionPair DimPair; typedef
H A Dcxx11_tensor_contraction.cpp17 typedef Tensor<float, 1>::DimensionPair DimPair; typedef
H A Dcxx11_tensor_of_float16_gpu.cu295 typedef Tensor<float, 2>::DimensionPair DimPair; in test_gpu_contractions() typedef
H A Dcxx11_tensor_gpu.cu297 typedef Tensor<float, 1>::DimensionPair DimPair; in test_gpu_contraction() typedef
/aosp_15_r20/external/eigen/bench/tensors/
H A Dtensor_contract_sycl_bench.cc77 typedef typename Tensor<T, 2>::DimensionPair DimPair; in contraction() typedef
131 typedef typename Tensor<T, 2>::DimensionPair DimPair; in contractionRowMajor() typedef
182 typedef typename Tensor<T, 2>::DimensionPair DimPair; in contractionAT() typedef
234 typedef typename Tensor<T, 2>::DimensionPair DimPair; in contractionBT() typedef
286 typedef typename Tensor<T, 2>::DimensionPair DimPair; in contractionABT() typedef
H A Dtensor_benchmarks.h541 typedef typename Tensor<T, 2, Layout>::DimensionPair DimPair; in contraction() typedef
/aosp_15_r20/external/eigen/bench/btl/libs/tensors/
H A Dtensor_interface.hh77 typedef typename Eigen::Tensor<real_type, 1>::DimensionPair DimPair; in matrix_matrix_product() typedef
83 typedef typename Eigen::Tensor<real_type, 1>::DimensionPair DimPair; in matrix_vector_product() typedef
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_matmul.cc60 typedef typename Eigen::Tensor<T, 2>::DimensionPair DimPair; in MatMul() typedef
98 typedef typename Eigen::Tensor<T, 2>::DimensionPair DimPair; in MatMul_Batch() typedef
H A Druntime_single_threaded_matmul.cc53 typedef typename Eigen::Tensor<T, 2>::DimensionPair DimPair; in MatMul() typedef
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_lrn_op.cc304 typedef typename Eigen::Tensor<T, 1, Eigen::RowMajor>::DimensionPair DimPair; typedef in tensorflow::MklLRNOp
651 typedef typename Eigen::Tensor<T, 1, Eigen::RowMajor>::DimensionPair DimPair; typedef in tensorflow::MklLRNGradOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dlrn_op.cc128 typedef typename Eigen::Tensor<T, 1, Eigen::RowMajor>::DimensionPair DimPair; typedef
H A Deigen_spatial_convolutions_test.cc1285 typedef Tensor<float, 1>::DimensionPair DimPair; in TEST() typedef