Searched refs:getScalarDataWithDefault (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 173 params_.activation = static_cast<TfLiteFusedActivation>(getScalarDataWithDefault<int32_t>( in BidirectionalSequenceLSTM()
178 params_.cell_clip = getScalarDataWithDefault<float>(clipOperand, 0.0f); in BidirectionalSequenceLSTM()
179 params_.proj_clip = getScalarDataWithDefault<float>(projOperand, 0.0f); in BidirectionalSequenceLSTM()
182 static_cast<float>(getScalarDataWithDefault<_Float16>(clipOperand, 0.0f)); in BidirectionalSequenceLSTM()
184 static_cast<float>(getScalarDataWithDefault<_Float16>(projOperand, 0.0f)); in BidirectionalSequenceLSTM()
187 params_.merge_outputs = getScalarDataWithDefault<bool>(mergeOutputsOperand, false); in BidirectionalSequenceLSTM()
189 params_.time_major = getScalarDataWithDefault<bool>(timeMajorOperand, false); in BidirectionalSequenceLSTM()
|
| D | SVDF.cpp | 42 params_.rank_ = getScalarDataWithDefault<int>(rankOperand, 0); in SVDF()
44 params_.activation_ = static_cast<TfLiteFusedActivation>(getScalarDataWithDefault<int>( in SVDF()
|
| D | LSTM.cpp | 87 params_.activation = static_cast<TfLiteFusedActivation>(getScalarDataWithDefault<int32_t>( in LSTMCell()
93 params_.cell_clip = getScalarDataWithDefault<float>(cellClipOperand, 0.0f); in LSTMCell()
94 params_.proj_clip = getScalarDataWithDefault<float>(projClipOperand, 0.0f); in LSTMCell()
97 static_cast<float>(getScalarDataWithDefault<_Float16>(cellClipOperand, 0.0f)); in LSTMCell()
99 static_cast<float>(getScalarDataWithDefault<_Float16>(projClipOperand, 0.0f)); in LSTMCell()
|
| /frameworks/ml/nn/common/include/ |
| D | CpuExecutor.h | 254 T getScalarDataWithDefault(const RunTimeOperandInfo& info, T defaultValue) { in getScalarDataWithDefault() function
|