Home | History | Annotate | Download | only in Tensor

Lines Matching refs:PacketAccess

273 template <typename Self, typename Op, typename OutputType, bool PacketAccess, typename Enabled = void>
281 template <typename Self, typename Op, typename OutputType, bool PacketAccess>
283 Self, Op, OutputType, PacketAccess,
351 (internal::is_same<typename Self::CoeffReturnType, Eigen::half>::value && reducer_traits<Op, GpuDevice>::PacketAccess));
367 FullReductionLauncher<Self, Op, OutputType, reducer_traits<Op, GpuDevice>::PacketAccess>::run(self, reducer, device, output, num_coeffs);
539 template <typename Self, typename Op, typename OutputType, bool PacketAccess, typename Enabled = void>
548 template <typename Self, typename Op, typename OutputType, bool PacketAccess>
550 Self, Op, OutputType, PacketAccess,
641 (internal::is_same<typename Self::CoeffReturnType, Eigen::half>::value && reducer_traits<Op, GpuDevice>::PacketAccess));
661 return InnerReductionLauncher<Self, Op, OutputType, reducer_traits<Op, GpuDevice>::PacketAccess>::run(self, reducer, device, output, num_coeffs_to_reduce, num_preserved_vals);