Searched refs:d_input (Results 1 – 3 of 3) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_cuda.cu | 331 float* d_input; in test_cuda_convolution_1d() local
334 cudaMalloc((void**)(&d_input), input_bytes); in test_cuda_convolution_1d()
338 cudaMemcpy(d_input, input.data(), input_bytes, cudaMemcpyHostToDevice); in test_cuda_convolution_1d()
344 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_input(d_input, 74,37,11,137); in test_cuda_convolution_1d()
367 cudaFree(d_input); in test_cuda_convolution_1d()
384 float* d_input; in test_cuda_convolution_inner_dim_col_major_1d() local
387 cudaMalloc((void**)(&d_input), input_bytes); in test_cuda_convolution_inner_dim_col_major_1d()
391 cudaMemcpy(d_input, input.data(), input_bytes, cudaMemcpyHostToDevice); in test_cuda_convolution_inner_dim_col_major_1d()
397 Eigen::TensorMap<Eigen::Tensor<float, 4, ColMajor> > gpu_input(d_input,74,9,11,7); in test_cuda_convolution_inner_dim_col_major_1d()
420 cudaFree(d_input); in test_cuda_convolution_inner_dim_col_major_1d()
[all …]
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | inc_op_kernel.cu.cc | 37 void IncrementKernel(const float* d_input, float inc, float* d_output, in IncrementKernel() argument
43 stream, d_input, inc, d_output, count)); in IncrementKernel()
|
| D | inc_op_kernel.h | 26 void IncrementKernel(const float* d_input, float inc, float* d_output,
|