Home
last modified time | relevance | path

Searched refs:CudaStreamDevice (Results 1 – 14 of 14) sorted by relevance

/external/eigen/bench/tensors/
Dtensor_benchmarks_fp16_gpu.cu13 Eigen::CudaStreamDevice stream; \
43 Eigen::CudaStreamDevice stream; \
62 Eigen::CudaStreamDevice stream; \
Dtensor_benchmarks_gpu.cu13 Eigen::CudaStreamDevice stream; \
43 Eigen::CudaStreamDevice stream; \
62 Eigen::CudaStreamDevice stream; \
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h99 class CudaStreamDevice : public StreamInterface { in initializeDeviceProp()
102 CudaStreamDevice() : stream_(&default_stream), scratch_(NULL), semaphore_(NULL) { in initializeDeviceProp() function
107CudaStreamDevice(int device) : stream_(&default_stream), device_(device), scratch_(NULL), semaphor… in initializeDeviceProp() function
114 CudaStreamDevice(const cudaStream_t* stream, int device = -1) in initializeDeviceProp()
129 virtual ~CudaStreamDevice() { in initializeDeviceProp()
/external/eigen/unsupported/test/
Dcxx11_tensor_cuda.cu38 Eigen::CudaStreamDevice stream; in test_cuda_nullary()
89 Eigen::CudaStreamDevice stream; in test_cuda_elementwise_small()
144 Eigen::CudaStreamDevice stream; in test_cuda_elementwise()
186 Eigen::CudaStreamDevice stream; in test_cuda_props()
224 Eigen::CudaStreamDevice stream; in test_cuda_reduction()
284 Eigen::CudaStreamDevice stream; in test_cuda_contraction()
341 Eigen::CudaStreamDevice stream; in test_cuda_convolution_1d()
394 Eigen::CudaStreamDevice stream; in test_cuda_convolution_inner_dim_col_major_1d()
447 Eigen::CudaStreamDevice stream; in test_cuda_convolution_inner_dim_row_major_1d()
501 Eigen::CudaStreamDevice stream; in test_cuda_convolution_2d()
[all …]
Dcxx11_tensor_random_cuda.cu33 Eigen::CudaStreamDevice stream; in test_cuda_random_uniform()
58 Eigen::CudaStreamDevice stream; in test_cuda_random_normal()
Dcxx11_tensor_reduction_cuda.cu25 Eigen::CudaStreamDevice stream; in test_full_reductions()
73 Eigen::CudaStreamDevice stream; in test_first_dim_reductions()
113 Eigen::CudaStreamDevice stream; in test_last_dim_reductions()
Dcxx11_tensor_complex_cuda.cu40 Eigen::CudaStreamDevice stream; in test_cuda_nullary()
76 Eigen::CudaStreamDevice stream; in test_cuda_sum_reductions()
113 Eigen::CudaStreamDevice stream; in test_cuda_product_reductions()
Dcxx11_tensor_of_float16_cuda.cu26 Eigen::CudaStreamDevice stream; in test_cuda_numext()
66 Eigen::CudaStreamDevice stream; in test_cuda_conversion()
101 Eigen::CudaStreamDevice stream; in test_cuda_unary()
138 Eigen::CudaStreamDevice stream; in test_cuda_elementwise()
180 Eigen::CudaStreamDevice stream; in test_cuda_trancendental()
268 Eigen::CudaStreamDevice stream; in test_cuda_contractions()
323 Eigen::CudaStreamDevice stream; in test_cuda_reductions()
380 Eigen::CudaStreamDevice stream; in test_cuda_full_reductions()
430 Eigen::CudaStreamDevice stream; in test_cuda_forced_evals()
Dcxx11_tensor_argmax_cuda.cu46 Eigen::CudaStreamDevice stream; in test_cuda_simple_argmax()
108 Eigen::CudaStreamDevice stream; in test_cuda_argmax_dim()
197 Eigen::CudaStreamDevice stream; in test_cuda_argmin_dim()
Dcxx11_tensor_contract_cuda.cu57 Eigen::CudaStreamDevice stream; in test_cuda_contraction()
121 Eigen::CudaStreamDevice stream; in test_scalar()
Dcxx11_tensor_cast_float16_cuda.cu25 Eigen::CudaStreamDevice stream; in test_cuda_conversion()
Dcxx11_tensor_scan_cuda.cu46 Eigen::CudaStreamDevice stream; in test_cuda_cumsum()
Dcxx11_tensor_complex_cwise_ops_cuda.cu34 Eigen::CudaStreamDevice stream; in test_cuda_complex_cwise_ops()
Dcxx11_tensor_device.cu107 Eigen::CudaStreamDevice stream_;