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

  /external/tensorflow/tensorflow/contrib/lite/kernels/
activation_functor.h 30 class ActivationFunctor {
32 explicit ActivationFunctor(TfLiteFusedActivation act) : act_(act) {}
  /frameworks/ml/nn/common/include/
ActivationFunctor.h 35 class ActivationFunctor {
37 explicit ActivationFunctor(ActivationFn act) : act_(act) {}
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
portable_tensor_utils.cc 110 auto sigmoid_func = ActivationFunctor(kTfLiteActSigmoid);
119 auto activation_func = ActivationFunctor(activation);
  /frameworks/ml/nn/common/operations/
RNN.cpp 132 (ActivationFunctor(activation_))(output_ptr_batch[o]);

Completed in 435 milliseconds