HomeSort by relevance Sort by last modified time
    Searched defs:ReduceFunctor (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
reduction_ops.h 69 struct ReduceFunctor {
reduction_ops_common.h 182 typedef functor::ReduceFunctor<Device, Reducer> Functor;
269 struct ReduceFunctor<CPUDevice, Reducer>
273 struct ReduceFunctor<SYCLDevice, Reducer>
reduction_gpu_kernels.cu.h 815 struct ReduceFunctor<GPUDevice, Reducer> {
823 struct ReduceFunctor<GPUDevice, Eigen::internal::SumReducer<T>> {
843 struct ReduceFunctor<GPUDevice, Eigen::internal::MeanReducer<T>> {
879 struct ReduceFunctor<GPUDevice, Eigen::internal::MeanReducer<Eigen::half>> {
    [all...]

Completed in 58 milliseconds