| 33 | |
| 34 | template <typename Device, typename T, int NDims> |
| 35 | struct Split { |
| 36 | void operator()(const Device& d, typename TTypes<T, NDims>::Tensor output, |
| 37 | typename TTypes<T, NDims>::ConstTensor input, |
| 38 | const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices, |
| 39 | const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_sizes); |
| 40 | }; |
| 41 | |
| 42 | template <typename T, int NDims> |
| 43 | struct Split<Eigen::ThreadPoolDevice, T, NDims> { |
no outgoing calls