Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:kActivationParam (Results 1 - 9 of 9) sorted by null

/frameworks/ml/nn/common/operations/
RNN.h	54 static constexpr int kActivationParam = 5;
SVDF.h	63 static constexpr int kActivationParam = 6;
LSTM.h	94 static constexpr int kActivationParam = 20;
RNN.cpp	34 getScalarData<int32_t>(operands[operation.inputs[kActivationParam]]));
RNNTest.cpp	241 ASSERT_EQ(execution.setInput(RNN::kActivationParam, &activation_,
SVDF.cpp	54 *GetInput(operation, operands, kActivationParam)));
SVDFTest.cpp	222 ASSERT_EQ(execution.setInput(SVDF::kActivationParam, &activation,
LSTM.cpp	71 *GetInput(operation, operands, kActivationParam))); 86 .activation_ = static_cast<ActivationFn>(getScalarData<int32_t>(*GetInput(operation, operands, LSTMCell::kActivationParam))),
LSTMTest.cpp	236 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam,     [all...]

Completed in 122 milliseconds