Searched defs:GPUContext (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_device.cu | 69 struct GPUContext { struct
70 …GPUContext(const Eigen::TensorMap<Eigen::Tensor<float, 3> >& in1, Eigen::TensorMap<Eigen::Tensor<f… in GPUContext() function
83 ~GPUContext() { in ~GPUContext() argument
89 const Eigen::GpuDevice& device() const { return gpu_device_; } in device()
91 const Eigen::TensorMap<Eigen::Tensor<float, 3> >& in1() const { return in1_; } in in1()
92 const Eigen::TensorMap<Eigen::Tensor<float, 3> >& in2() const { return in2_; } in in2()
93 Eigen::TensorMap<Eigen::Tensor<float, 3> >& out() { return out_; } in out()
94 …<float, 1> > kernel1d() const { return Eigen::TensorMap<Eigen::Tensor<float, 1> >(kernel_1d_, 2); } in kernel1d()
95 …oat, 2> > kernel2d() const { return Eigen::TensorMap<Eigen::Tensor<float, 2> >(kernel_2d_, 2, 2); } in kernel2d()
96 …, 3> > kernel3d() const { return Eigen::TensorMap<Eigen::Tensor<float, 3> >(kernel_3d_, 2, 2, 2); } in kernel3d()
[all …]
|