| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | quant_lstm_sup_test.cc | 260 std::vector<int32_t> cell_bias; in TEST() local
283 std::vector<int32_t> cell_bias; in TEST() local
306 std::vector<int32_t> cell_bias; in TEST() local
329 std::vector<int32_t> cell_bias; in TEST() local
|
| H A D | quant_lstm_sup.cc | 132 std::vector<int32_t>* cell_bias, in DecomposeBiasTensor()
|
| H A D | nnapi_delegate.cc | 4128 std::vector<int32_t> cell_bias; in Map() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLLSTMLayer.cpp | 63 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias, in configure()
76 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias, in configure()
403 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_… in validate()
|
| H A D | CLLSTMLayerQuantized.cpp | 63 …ICLTensor *input_gate_bias, const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const I… in configure()
75 …ICLTensor *input_gate_bias, const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const I… in configure()
283 …orInfo *input_gate_bias, const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const … in validate()
|
| H A D | CLQLSTMLayer.cpp | 162 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias, in configure()
175 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias, in configure()
606 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NELSTMLayer.cpp | 60 … const ITensor *forget_gate_bias, const ITensor *cell_bias, const ITensor *output_gate_bias, in configure()
378 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_… in validate()
|
| H A D | NELSTMLayerQuantized.cpp | 63 …onst ITensor *input_gate_bias, const ITensor *forget_gate_bias, const ITensor *cell_bias, const IT… in configure()
251 …orInfo *input_gate_bias, const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const … in validate()
|
| H A D | NEQLSTMLayer.cpp | 217 … const ITensor *forget_gate_bias, const ITensor *cell_bias, const ITensor *output_gate_bias, in configure()
678 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_… in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 362 const TfLiteTensor* cell_bias, const TfLiteTensor* output_gate_bias, in EvalFloat()
593 const TfLiteTensor* cell_bias = in lstm_eval() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | LSTMLayerFixture.h | 95 TensorType cell_bias = create_tensor<TensorType>(cell_bias_shape, data_type); in compute_target() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 144 const RunTimeOperandInfo* forget_gate_bias, const RunTimeOperandInfo* cell_bias, in CheckInputTensorDimensions()
|