HomeSort by relevance Sort by last modified time
    Searched refs:reducer_traits (Results 1 - 2 of 2) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorFunctors.h 88 struct reducer_traits { struct in namespace:Eigen::internal
133 struct reducer_traits<SumReducer<T>, Device> { struct in namespace:Eigen::internal
187 struct reducer_traits<MeanReducer<T>, Device> { struct in namespace:Eigen::internal
254 struct reducer_traits<MaxReducer<T>, Device> { struct in namespace:Eigen::internal
295 struct reducer_traits<MinReducer<T>, Device> { struct in namespace:Eigen::internal
340 struct reducer_traits<ProdReducer<T>, Device> { struct in namespace:Eigen::internal
365 struct reducer_traits<AndReducer, Device> { struct in namespace:Eigen::internal
389 struct reducer_traits<OrReducer, Device> { struct in namespace:Eigen::internal
415 struct reducer_traits<ArgMaxTupleReducer<T>, Device> { struct in namespace:Eigen::internal
440 struct reducer_traits<ArgMinTupleReducer<T>, Device> struct in namespace:Eigen::internal
    [all...]
TensorReductionCuda.h 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);
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);

Completed in 159 milliseconds