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

  /frameworks/ml/nn/common/include/
ActivationFunctor.h 25 enum ActivationFn {
37 explicit ActivationFunctor(ActivationFn act) : act_(act) {}
60 ActivationFn act_;
  /frameworks/ml/nn/common/operations/
RNN.h 60 ActivationFn activation_;
RNN.cpp 33 activation_ = static_cast<ActivationFn>(
LSTMTest.cpp 86 activation_(ActivationFn::kActivationTanh),
    [all...]

Completed in 3100 milliseconds