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

  /external/tensorflow/tensorflow/core/kernels/
softmax_op.cc 46 struct SoftmaxFunctor<CPUDevice, T> : SoftmaxFunctorBase<CPUDevice, T> {};
50 struct SoftmaxFunctor<SYCLDevice, T> : SoftmaxFunctorBase<SYCLDevice, T> {};
69 functor::SoftmaxFunctor<Device, T> functor;
softmax_op_functor.h 28 struct SoftmaxFunctor {
38 // Eigen code implementing SoftmaxFunctor::operator() or

Completed in 59 milliseconds