| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | bidirectional_sequence_lstm_merge_outputs.model.cpp | 64 auto proj_clip = model->addOperand(&type7); in CreateModel() local
82 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel()
87 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel()
161 auto proj_clip = model->addOperand(&type7); in CreateModel_dynamic_output_shape() local
179 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel_dynamic_output_shape()
184 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel_dynamic_output_shape()
|
| D | bidirectional_sequence_lstm_float16_batch_major_merge_outputs.model.cpp | 64 auto proj_clip = model->addOperand(&type7); in CreateModel() local
82 model->setOperandValue(proj_clip, proj_clip_init, sizeof(_Float16) * 1); in CreateModel()
87 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel()
161 auto proj_clip = model->addOperand(&type7); in CreateModel_dynamic_output_shape() local
179 model->setOperandValue(proj_clip, proj_clip_init, sizeof(_Float16) * 1); in CreateModel_dynamic_output_shape()
184 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel_dynamic_output_shape()
|
| D | bidirectional_sequence_lstm_cifg_peephole.model.cpp | 64 auto proj_clip = model->addOperand(&type7); in CreateModel() local
83 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel()
88 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel()
162 auto proj_clip = model->addOperand(&type7); in CreateModel_dynamic_output_shape() local
181 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel_dynamic_output_shape()
186 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel_dynamic_output_shape()
|
| D | bidirectional_sequence_lstm_float16_batch_major.model.cpp | 64 auto proj_clip = model->addOperand(&type7); in CreateModel() local
83 model->setOperandValue(proj_clip, proj_clip_init, sizeof(_Float16) * 1); in CreateModel()
88 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel()
162 auto proj_clip = model->addOperand(&type7); in CreateModel_dynamic_output_shape() local
181 model->setOperandValue(proj_clip, proj_clip_init, sizeof(_Float16) * 1); in CreateModel_dynamic_output_shape()
186 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel_dynamic_output_shape()
|
| D | bidirectional_sequence_lstm_aux_input.model.cpp | 64 auto proj_clip = model->addOperand(&type7); in CreateModel() local
83 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel()
88 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel()
162 auto proj_clip = model->addOperand(&type7); in CreateModel_dynamic_output_shape() local
181 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel_dynamic_output_shape()
186 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel_dynamic_output_shape()
|
| D | bidirectional_sequence_lstm_norm_fw_output.model.cpp | 67 auto proj_clip = model->addOperand(&type10); in CreateModel() local
86 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel()
91 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel()
168 auto proj_clip = model->addOperand(&type10); in CreateModel_dynamic_output_shape() local
187 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel_dynamic_output_shape()
192 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel_dynamic_output_shape()
|
| D | bidirectional_sequence_lstm.model.cpp | 64 auto proj_clip = model->addOperand(&type7); in CreateModel() local
83 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel()
88 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel()
162 auto proj_clip = model->addOperand(&type7); in CreateModel_dynamic_output_shape() local
181 model->setOperandValue(proj_clip, proj_clip_init, sizeof(float) * 1); in CreateModel_dynamic_output_shape()
186 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel_dynamic_output_shape()
|
| D | bidirectional_sequence_lstm_float16_batch_major_aux_input.model.cpp | 64 auto proj_clip = model->addOperand(&type7); in CreateModel() local
83 model->setOperandValue(proj_clip, proj_clip_init, sizeof(_Float16) * 1); in CreateModel()
88 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel()
162 auto proj_clip = model->addOperand(&type7); in CreateModel_dynamic_output_shape() local
181 model->setOperandValue(proj_clip, proj_clip_init, sizeof(_Float16) * 1); in CreateModel_dynamic_output_shape()
186 …to_cell_weights, bw_aux_input_to_output_weights, activation, cell_clip, proj_clip, merge_outputs, … in CreateModel_dynamic_output_shape()
|
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | bidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py | 212 proj_clip = Float16Scalar("proj_clip", 0.0)
266 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_merge_outputs.mod.py | 211 proj_clip = Float32Scalar("proj_clip", 0.0)
265 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_float16_batch_major.mod.py | 212 proj_clip = Float16Scalar("proj_clip", 0.0)
266 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm.mod.py | 212 proj_clip = Float32Scalar("proj_clip", 0.0)
266 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_cifg_peephole.mod.py | 212 proj_clip = Float32Scalar("proj_clip", 0.0)
266 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_aux_input.mod.py | 214 proj_clip = Float32Scalar("proj_clip", 0.0)
268 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_norm_fw_output.mod.py | 213 proj_clip = Float32Scalar("proj_clip", 0.0)
267 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py | 215 proj_clip = Float16Scalar("proj_clip", 0.0)
269 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| /frameworks/ml/nn/common/operations/ |
| D | LayerNormLSTMTest.cpp | 89 bool use_projection_bias, float cell_clip, float proj_clip, in LayerNormLSTMOpModel() argument
99 proj_clip_(proj_clip) { in LayerNormLSTMOpModel()
|
| D | LSTM.cpp | 85 params_.proj_clip = getScalarData<float>(*GetInput(operation, operands, kProjClipParam)); in LSTMCell()
89 params_.proj_clip = static_cast<float>( in LSTMCell()
147 NN_CHECK(params->proj_clip >= 0); in CheckInputTensorDimensions()
941 if (params.proj_clip > 0.0) { in LSTMStep()
942 tflite::tensor_utils::ClipVector(output_buffer, n_batch * n_output, params.proj_clip, in LSTMStep()
|
| D | LSTMTest.cpp | 79 bool use_projection_bias, float cell_clip, float proj_clip, in LSTMOpModel() argument
87 cell_clip_(cell_clip), proj_clip_(proj_clip) { in LSTMOpModel()
|
| D | LSTM.h | 33 float proj_clip; member
|
| D | BidirectionalSequenceLSTM.cpp | 143 params_.proj_clip = getScalarData<float>(*GetInput(operation, operands, kProjClipParam)); in BidirectionalSequenceLSTM()
147 params_.proj_clip = static_cast<float>( in BidirectionalSequenceLSTM()
|
| D | UnidirectionalSequenceLSTM.cpp | 100 params.proj_clip = static_cast<float>(context->getInputValue<T>(kProjClipParam)); in getLSTMParams()
|