| 73 | template <typename Scalar, bool IsComplex = true> |
| 74 | struct ParallelMatMulKernel { |
| 75 | static void Conjugate(const OpKernelContext* context, Tensor* out) { |
| 76 | const Eigen::ThreadPoolDevice d = context->eigen_cpu_device(); |
| 77 | auto z = out->tensor<Scalar, 3>(); |
| 78 | z.device(d) = z.conjugate(); |
| 79 | } |
| 80 | |
| 81 | static void Run(const OpKernelContext* context, const Tensor& in_x, |
| 82 | const Tensor in_y, bool adj_x, bool adj_y, bool trans_x, |
nothing calls this directly
no test coverage detected