Searched refs:gpuMalloc (Results 1 – 10 of 10) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_gpu.cu | 34 gpuMalloc((void**)(&d_in1), tensor_bytes); in test_gpu_nullary()
35 gpuMalloc((void**)(&d_in2), tensor_bytes); in test_gpu_nullary()
83 gpuMalloc((void**)(&d_in1), in1_bytes); in test_gpu_elementwise_small()
84 gpuMalloc((void**)(&d_in2), in2_bytes); in test_gpu_elementwise_small()
85 gpuMalloc((void**)(&d_out), out_bytes); in test_gpu_elementwise_small()
136 gpuMalloc((void**)(&d_in1), in1_bytes); in test_gpu_elementwise()
137 gpuMalloc((void**)(&d_in2), in2_bytes); in test_gpu_elementwise()
138 gpuMalloc((void**)(&d_in3), in3_bytes); in test_gpu_elementwise()
139 gpuMalloc((void**)(&d_out), out_bytes); in test_gpu_elementwise()
182 gpuMalloc((void**)(&d_in1), in1_bytes); in test_gpu_props()
[all …]
|
| D | cxx11_tensor_contract_gpu.cu | 49 gpuMalloc((void**)(&d_t_left), t_left_bytes); in test_gpu_contraction()
50 gpuMalloc((void**)(&d_t_right), t_right_bytes); in test_gpu_contraction()
51 gpuMalloc((void**)(&d_t_result), t_result_bytes); in test_gpu_contraction()
113 gpuMalloc((void**)(&d_t_left), t_left_bytes); in test_scalar()
114 gpuMalloc((void**)(&d_t_right), t_right_bytes); in test_scalar()
115 gpuMalloc((void**)(&d_t_result), t_result_bytes); in test_scalar()
|
| D | cxx11_tensor_argmax_gpu.cu | 39 gpuMalloc((void**)(&d_in), in_bytes); in test_gpu_simple_argmax()
40 gpuMalloc((void**)(&d_out_max), out_bytes); in test_gpu_simple_argmax()
41 gpuMalloc((void**)(&d_out_min), out_bytes); in test_gpu_simple_argmax()
102 gpuMalloc((void**)(&d_in), in_bytes); in test_gpu_argmax_dim()
103 gpuMalloc((void**)(&d_out), out_bytes); in test_gpu_argmax_dim()
191 gpuMalloc((void**)(&d_in), in_bytes); in test_gpu_argmin_dim()
192 gpuMalloc((void**)(&d_out), out_bytes); in test_gpu_argmin_dim()
|
| D | cxx11_tensor_random_gpu.cu | 29 gpuMalloc((void**)(&d_out), out_bytes); in test_gpu_random_uniform()
54 gpuMalloc((void**)(&d_out), out_bytes); in test_gpu_random_normal()
|
| D | cxx11_tensor_scan_gpu.cu | 40 gpuMalloc((void**)(&d_t_input), t_input_bytes); in test_gpu_cumsum()
41 gpuMalloc((void**)(&d_t_result), t_result_bytes); in test_gpu_cumsum()
|
| D | cxx11_tensor_device.cu | 70 assert(gpuMalloc((void**)(&kernel_1d_), 2*sizeof(float)) == gpuSuccess); in GPUContext()
74 assert(gpuMalloc((void**)(&kernel_2d_), 4*sizeof(float)) == gpuSuccess); in GPUContext()
78 assert(gpuMalloc((void**)(&kernel_3d_), 8*sizeof(float)) == gpuSuccess); in GPUContext()
285 gpuMalloc((void**)(&d_in1), in1_bytes); in test_gpu()
286 gpuMalloc((void**)(&d_in2), in2_bytes); in test_gpu()
287 gpuMalloc((void**)(&d_out), out_bytes); in test_gpu()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorGpuHipCudaDefines.h | 41 #define gpuMalloc hipMalloc macro
71 #define gpuMalloc cudaMalloc macro
|
| D | TensorGpuHipCudaUndefines.h | 26 #undef gpuMalloc
|
| D | TensorDeviceGpu.h | 168 err = gpuMalloc(&result, num_bytes); in allocate()
|
| /external/eigen/test/ |
| D | gpu_common.h | 50 gpuMalloc((void**)(&d_in), in_bytes); in run_on_gpu()
51 gpuMalloc((void**)(&d_out), out_bytes); in run_on_gpu()
|