Searched refs:kernel_1d_ (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_device.cu | 27 … in2, Eigen::Tensor<float, 3>& out) : in1_(in1), in2_(in2), out_(out), kernel_1d_(2), kernel_2d_(2… in CPUContext()
28 kernel_1d_(0) = 3.14f; in CPUContext()
29 kernel_1d_(1) = 2.7f; in CPUContext()
51 const Eigen::Tensor<float, 1>& kernel1d() const { return kernel_1d_; } in kernel1d()
60 Eigen::Tensor<float, 1> kernel_1d_; member
71 assert(cudaMalloc((void**)(&kernel_1d_), 2*sizeof(float)) == cudaSuccess); in GPUContext()
73 …assert(cudaMemcpy(kernel_1d_, kernel_1d_val, 2*sizeof(float), cudaMemcpyHostToDevice) == cudaSucce… in GPUContext()
84 assert(cudaFree(kernel_1d_) == cudaSuccess); in ~GPUContext()
94 …<float, 1> > kernel1d() const { return Eigen::TensorMap<Eigen::Tensor<float, 1> >(kernel_1d_, 2); } in kernel1d()
103 float* kernel_1d_; member
|