Home
last modified time | relevance | path

Searched defs:input_to_cell_weights (Results 1 – 25 of 64) sorted by relevance

123

/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DLSTMLayerQuantized.cpp95 …auto input_to_cell_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE() local
234 …auto input_to_cell_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_CASE() local
474 …auto input_to_cell_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLLSTMLayer.cpp61 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor … in configure()
74 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor … in configure()
401 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso… in validate()
H A DCLLSTMLayerQuantized.cpp61 …input_weights, const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, c… in configure()
73 …input_weights, const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, c… in configure()
281 …ut_weights, const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, … in validate()
H A DCLQLSTMLayer.cpp160 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor … in configure()
173 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor … in configure()
604 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso… in validate()
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/
H A DLSTMLayerQuantized.cpp102 …auto input_to_cell_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE() local
241 …auto input_to_cell_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_CASE() local
481 …auto input_to_cell_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm.cc103 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_16() local
446 const TfLiteTensor* input_to_cell_weights; in CheckInputTensorDimensions() local
733 const TfLiteTensor* input_to_cell_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
1236 const TfLiteTensor* input_to_cell_weights; in Eval() local
H A Dlstm.cc188 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_16() local
476 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_8() local
887 const TfLiteTensor* input_to_cell_weights; in CheckInputTensorDimensions() local
1171 const TfLiteTensor* input_to_cell_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
1658 const TfLiteTensor* input_to_cell_weights = in Prepare() local
1842 const TfLiteTensor* input_to_cell_weights; in Eval() local
H A Dlstm_test.cc1612 const std::vector<float> input_to_cell_weights = { in TEST() local
1774 const std::vector<float> input_to_cell_weights = { in TEST() local
1947 const std::vector<float> input_to_cell_weights = { in TEST() local
2137 const std::vector<float>& input_to_cell_weights, in HybridSparseLSTMOpModel()
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dquantized_lstm.mod.py35 input_to_cell_weights = Input("inputToCellWeights", InputWeightsType) variable
124 input_to_cell_weights = Parameter("inputToCellWeights", InputWeightsType, variable
Dlayer_norm_lstm.mod.py33 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", variable
205 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", variable
Dlstm3_state2_float16.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_in… variable
Dlstm2_state_float16.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_in… variable
Dlstm2_float16.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_in… variable
Dlstm3_float16.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_in… variable
Dlstm3_state3_float16.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_in… variable
Dlstm2_state2_float16.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_in… variable
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNELSTMLayer.cpp58 …const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input… in configure()
376 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso… in validate()
H A DNELSTMLayerQuantized.cpp61 …to_input_weights, const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, co… in configure()
249 …ut_weights, const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, … in validate()
H A DNEQLSTMLayer.cpp215 …const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input… in configure()
676 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso… in validate()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc343 const TfLiteTensor* input_to_cell_weights, in EvalFloat()
555 const TfLiteTensor* input_to_cell_weights = GetInput( in lstm_eval() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc267 const TfLiteTensor* input_to_cell_weights, in EvalCalibration()
481 const TfLiteTensor* input_to_cell_weights; in lstm_eval() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
Dlstm3_state.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_in… variable
Dlstm2_state.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_in… variable
Dlstm2.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_in… variable
Dlstm3_state3.mod.py31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_in… variable

123