Searched defs:CPUContext (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_device.cu | 25 struct CPUContext { struct
26 …CPUContext(const Eigen::Tensor<float, 3>& in1, Eigen::Tensor<float, 3>& in2, Eigen::Tensor<float, … in CPUContext() function
45 const Eigen::DefaultDevice& device() const { return cpu_device_; } in device()
47 const Eigen::Tensor<float, 3>& in1() const { return in1_; } in in1()
48 const Eigen::Tensor<float, 3>& in2() const { return in2_; } in in2()
49 Eigen::Tensor<float, 3>& out() { return out_; } in out()
50 const Eigen::Tensor<float, 1>& kernel1d() const { return kernel_1d_; } in kernel1d()
51 const Eigen::Tensor<float, 2>& kernel2d() const { return kernel_2d_; } in kernel2d()
52 const Eigen::Tensor<float, 3>& kernel3d() const { return kernel_3d_; } in kernel3d()
55 const Eigen::Tensor<float, 3>& in1_;
[all …]
|