Home
last modified time | relevance | path

Searched refs:cell_clip_ (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSTM.cpp71 params_.cell_clip_ = getScalarData<float>(*GetInput(operation, operands, kCellClipParam)); in LSTMCell()
87 .cell_clip_ = getScalarData<float>(*GetInput(operation, operands, LSTMCell::kCellClipParam)), in CheckInputTensorDimensions()
94 NN_CHECK(params.cell_clip_ >= 0); in CheckInputTensorDimensions()
414 if (params_.cell_clip_ > 0.0) { in Eval()
416 params_.cell_clip_, GetBuffer<float>(cell_state_out_)); in Eval()
DLSTMTest.cpp87 cell_clip_(cell_clip), proj_clip_(proj_clip) { in LSTMOpModel()
240 &cell_clip_, sizeof(cell_clip_)), in Invoke()
261 const float cell_clip_; member in android::nn::wrapper::LSTMOpModel
DLSTM.h41 float cell_clip_; member