Searched refs:in1_bytes (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_cuda.cu | 75 std::size_t in1_bytes = in1.size() * sizeof(float); in test_cuda_elementwise_small() local
82 cudaMalloc((void**)(&d_in1), in1_bytes); in test_cuda_elementwise_small()
86 cudaMemcpy(d_in1, in1.data(), in1_bytes, cudaMemcpyHostToDevice); in test_cuda_elementwise_small()
126 std::size_t in1_bytes = in1.size() * sizeof(float); in test_cuda_elementwise() local
135 cudaMalloc((void**)(&d_in1), in1_bytes); in test_cuda_elementwise()
140 cudaMemcpy(d_in1, in1.data(), in1_bytes, cudaMemcpyHostToDevice); in test_cuda_elementwise()
176 std::size_t in1_bytes = in1.size() * sizeof(float); in test_cuda_props() local
181 cudaMalloc((void**)(&d_in1), in1_bytes); in test_cuda_props()
184 cudaMemcpy(d_in1, in1.data(), in1_bytes, cudaMemcpyHostToDevice); in test_cuda_props()
214 std::size_t in1_bytes = in1.size() * sizeof(float); in test_cuda_reduction() local
[all …]
|
| D | cxx11_tensor_device.cu | 279 std::size_t in1_bytes = in1.size() * sizeof(float); in test_gpu() local
286 cudaMalloc((void**)(&d_in1), in1_bytes); in test_gpu()
290 cudaMemcpy(d_in1, in1.data(), in1_bytes, cudaMemcpyHostToDevice); in test_gpu()
|