| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_thread_pool.cpp | 70 Tensor<float, 5, DataLayout> t_right(37, 31, 70, 2, 10); in test_multithread_contraction() local
74 t_right.setRandom(); in test_multithread_contraction()
82 MapXf m_right(t_right.data(), 1147, 1400); in test_multithread_contraction()
89 t_result.device(thread_pool_device) = t_left.contract(t_right, dims); in test_multithread_contraction()
110 Tensor<float, 2, DataLayout> t_right(32, 28*28); in test_contraction_corner_cases() local
114 t_right = (t_right.constant(-0.6f) + t_right.random()) * 2.0f; in test_contraction_corner_cases()
123 MapXf m_right(t_right.data(), 32, 28*28); in test_contraction_corner_cases()
130 t_result.device(thread_pool_device) = t_left.contract(t_right, dims); in test_contraction_corner_cases()
145 t_result.device(thread_pool_device) = t_left.contract(t_right, dims); in test_contraction_corner_cases()
157 t_right.resize(32, 4); in test_contraction_corner_cases()
[all …]
|
| D | cxx11_tensor_contract_cuda.cu | 34 Tensor<float, 2, DataLayout> t_right(k_size, n_size); in test_cuda_contraction() local
40 t_right.setRandom(); in test_cuda_contraction()
43 std::size_t t_right_bytes = t_right.size() * sizeof(float); in test_cuda_contraction()
55 cudaMemcpy(d_t_right, t_right.data(), t_right_bytes, cudaMemcpyHostToDevice); in test_cuda_contraction()
69 t_result = t_left.contract(t_right, dims); in test_cuda_contraction()
98 Tensor<float, 2, DataLayout> t_right(k_size, n_size); in test_scalar() local
104 t_right.setRandom(); in test_scalar()
107 std::size_t t_right_bytes = t_right.size() * sizeof(float); in test_scalar()
119 cudaMemcpy(d_t_right, t_right.data(), t_right_bytes, cudaMemcpyHostToDevice); in test_scalar()
132 t_result = t_left.contract(t_right, dims); in test_scalar()
|
| D | cxx11_tensor_contraction.cpp | 356 Tensor<float, 5, DataLayout> t_right(8, 31, 7, 20, 10); in test_large_contraction() local
360 t_right.setRandom(); in test_large_contraction()
364 t_right += t_right.constant(1.0f); in test_large_contraction()
368 MapXf m_right(t_right.data(), 248, 1400); in test_large_contraction()
375 t_result = t_left.contract(t_right, dims); in test_large_contraction()
388 Tensor<float, 1, DataLayout> t_right(50); in test_matrix_vector() local
392 t_right.setRandom(); in test_matrix_vector()
396 MapXf m_right(t_right.data(), 50, 1); in test_matrix_vector()
403 t_result = t_left.contract(t_right, dims); in test_matrix_vector()
416 Tensor<float, 2, DataLayout> t_right(1, 7); in test_tensor_vector() local
[all …]
|
| D | cxx11_tensor_of_complex.cpp | 73 Tensor<std::complex<float>, 5> t_right(8, 31, 7, 20, 10); in test_contractions() local
77 t_right.setRandom(); in test_contractions()
81 MapXcf m_right(t_right.data(), 248, 1400); in test_contractions()
89 t_result = t_left.contract(t_right, dims); in test_contractions()
|
| D | cxx11_tensor_cuda.cu | 263 Tensor<float, 5, DataLayout> t_right(Eigen::array<Eigen::DenseIndex, 5>(3, 31, 7, 20, 1)); in test_cuda_contraction() local
267 t_right.setRandom(); in test_cuda_contraction()
270 std::size_t t_right_bytes = t_right.size() * sizeof(float); in test_cuda_contraction()
282 cudaMemcpy(d_t_right, t_right.data(), t_right_bytes, cudaMemcpyHostToDevice); in test_cuda_contraction()
293 MapXf m_right(t_right.data(), 93, 140); in test_cuda_contraction()
|