| 41 | template <typename Device, typename In0, typename In1, typename Out, |
| 42 | typename DimPair> |
| 43 | void MatMul(const Device& d, Out out, In0 in0, In1 in1, |
| 44 | const DimPair& dim_pair) { |
| 45 | out.device(d) = in0.contract(in1, dim_pair); |
| 46 | } |
| 47 | |
| 48 | template <typename Device, typename T> |
| 49 | struct MatMulFunctor { |