Searched refs:use_cifg (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_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 | 88 bool use_cifg, bool use_peephole, bool use_projection_weights, in LayerNormLSTMOpModel() argument
93 use_cifg_(use_cifg), in LayerNormLSTMOpModel()
127 {n_batch, n_cell * (use_cifg ? 3 : 4)}, in LayerNormLSTMOpModel()
|
| D | LSTMTest.cpp | 79 bool use_cifg, bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
84 use_cifg_(use_cifg), in LSTMOpModel()
117 {n_batch, n_cell * (use_cifg ? 3 : 4)}, in LSTMOpModel()
|
| D | UnidirectionalSequenceLSTM.cpp | 55 params.use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in getLSTMParams()
294 const bool use_cifg = !hasTensor(context, kInputToInputWeightsTensor); in execute() local
295 const auto scratchSize = use_cifg ? 3 * cellStateSize : 4 * cellStateSize; in execute()
|
| D | BidirectionalSequenceLSTM.cpp | 300 if (params_.use_cifg) { in Prepare()
306 ((params_.use_cifg || !IsNullInput(fw_aux_input_to_input_weights_)) && in Prepare()
315 ((params_.use_cifg || !IsNullInput(bw_aux_input_to_input_weights_)) && in Prepare()
425 if (params_.use_cifg) { in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 35 bool use_cifg; member
|