Searched refs:in2_bytes (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_device.cu | 280 std::size_t in2_bytes = in2.size() * sizeof(float); in test_gpu() local
287 cudaMalloc((void**)(&d_in2), in2_bytes); in test_gpu()
291 cudaMemcpy(d_in2, in2.data(), in2_bytes, cudaMemcpyHostToDevice); in test_gpu()
|
| D | cxx11_tensor_cuda.cu | 76 std::size_t in2_bytes = in2.size() * sizeof(float); in test_cuda_elementwise_small() local
83 cudaMalloc((void**)(&d_in2), in2_bytes); in test_cuda_elementwise_small()
87 cudaMemcpy(d_in2, in2.data(), in2_bytes, cudaMemcpyHostToDevice); in test_cuda_elementwise_small()
127 std::size_t in2_bytes = in2.size() * sizeof(float); in test_cuda_elementwise() local
136 cudaMalloc((void**)(&d_in2), in2_bytes); in test_cuda_elementwise()
141 cudaMemcpy(d_in2, in2.data(), in2_bytes, cudaMemcpyHostToDevice); in test_cuda_elementwise()
|