Searched refs:use_cifg (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 209 params->use_cifg = IsNullInput(input_to_input_weights); in CheckInputTensorDimensions()
211 ((!IsNullInput(cell_to_input_weights) || params->use_cifg) && in CheckInputTensorDimensions()
228 if (params->use_cifg) { in CheckInputTensorDimensions()
281 if (params->use_cifg) { in CheckInputTensorDimensions()
390 if (params_.use_cifg) { in Prepare()
676 std::vector<float> scratch_buffer_float32(params.use_cifg ? 3 * batchSize * numCells in LSTMEvalFloat16()
800 if (params.use_cifg) { in LSTMStep()
813 if (!params.use_cifg) { in LSTMStep()
825 if (!params.use_cifg) { in LSTMStep()
834 if (!params.use_cifg) { in LSTMStep()
[all …]
|
| D | LayerNormLSTMTest.cpp | 87 bool use_cifg, bool use_peephole, bool use_projection_weights, in LayerNormLSTMOpModel() argument
92 use_cifg_(use_cifg), in LayerNormLSTMOpModel()
126 {n_batch, n_cell * (use_cifg ? 3 : 4)}, in LayerNormLSTMOpModel()
|
| D | LSTMTest.cpp | 80 bool use_cifg, bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
85 use_cifg_(use_cifg), in LSTMOpModel()
118 {n_batch, n_cell * (use_cifg ? 3 : 4)}, in LSTMOpModel()
|
| D | BidirectionalSequenceLSTM.cpp | 301 if (params_.use_cifg) { in Prepare()
307 ((params_.use_cifg || !IsNullInput(fw_aux_input_to_input_weights_)) && in Prepare()
316 ((params_.use_cifg || !IsNullInput(bw_aux_input_to_input_weights_)) && in Prepare()
426 if (params_.use_cifg) { in Prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 108 params.use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in getLSTMParams()
400 const bool use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in execute() local
401 const auto scratchSize = use_cifg ? 3 * cellStateSize : 4 * cellStateSize; in execute()
|
| D | LSTM.h | 36 bool use_cifg; member
|