Searched refs:kernel_3d_ (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_device.cu | 27 …at, 3>& out) : in1_(in1), in2_(in2), out_(out), kernel_1d_(2), kernel_2d_(2,2), kernel_3d_(2,2,2) { in CPUContext()
36 kernel_3d_(0,0,0) = 3.14f; in CPUContext()
37 kernel_3d_(0,1,0) = 2.7f; in CPUContext()
38 kernel_3d_(0,0,1) = 0.2f; in CPUContext()
39 kernel_3d_(0,1,1) = 7.0f; in CPUContext()
40 kernel_3d_(1,0,0) = -1.0f; in CPUContext()
41 kernel_3d_(1,1,0) = -0.3f; in CPUContext()
42 kernel_3d_(1,0,1) = -0.7f; in CPUContext()
43 kernel_3d_(1,1,1) = -0.5f; in CPUContext()
53 const Eigen::Tensor<float, 3>& kernel3d() const { return kernel_3d_; } in kernel3d()
[all …]
|