| /external/ComputeLibrary/tests/validation/NEON/ |
| D | LSTMLayerQuantized.cpp | 104 …auto recurrent_to_input_weights = create_tensor<Tensor>(recurrent_weights_shape, DataType::QASYMM… in TEST_SUITE() local
125 …&recurrent_to_input_weights, &recurrent_to_forget_weights, &recurrent_to_cell_weights, &recurrent_… in TEST_SUITE()
133 recurrent_to_input_weights.allocator()->allocate(); in TEST_SUITE()
165 fill_tensor(recurrent_to_input_weights, std::vector<uint8_t> { 162, 168, 7, 95, in TEST_SUITE()
243 …auto recurrent_to_input_weights = create_tensor<Tensor>(recurrent_weights_shape, DataType::QASYMM… in TEST_CASE() local
264 …&recurrent_to_input_weights, &recurrent_to_forget_weights, &recurrent_to_cell_weights, &recurrent_… in TEST_CASE()
272 recurrent_to_input_weights.allocator()->allocate(); in TEST_CASE()
320 …fill_tensor(recurrent_to_input_weights, std::vector<uint8_t> { 147, 112, 140, 103, 3, 255, 17, … in TEST_CASE()
483 …auto recurrent_to_input_weights = create_tensor<Tensor>(recurrent_weights_shape, DataType::QASYMM… in TEST_SUITE() local
504 …&recurrent_to_input_weights, &recurrent_to_forget_weights, &recurrent_to_cell_weights, &recurrent_… in TEST_SUITE()
[all …]
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | LSTMLayerQuantized.cpp | 97 …auto recurrent_to_input_weights = create_tensor<CLTensor>(recurrent_weights_shape, DataType::QASY… in TEST_SUITE() local
118 …&recurrent_to_input_weights, &recurrent_to_forget_weights, &recurrent_to_cell_weights, &recurrent_… in TEST_SUITE()
126 recurrent_to_input_weights.allocator()->allocate(); in TEST_SUITE()
158 fill_tensor(recurrent_to_input_weights, std::vector<uint8_t> { 162, 168, 7, 95, in TEST_SUITE()
236 …auto recurrent_to_input_weights = create_tensor<CLTensor>(recurrent_weights_shape, DataType::QASY… in TEST_CASE() local
257 …&recurrent_to_input_weights, &recurrent_to_forget_weights, &recurrent_to_cell_weights, &recurrent_… in TEST_CASE()
265 recurrent_to_input_weights.allocator()->allocate(); in TEST_CASE()
313 …fill_tensor(recurrent_to_input_weights, std::vector<uint8_t> { 147, 112, 140, 103, 3, 255, 17, … in TEST_CASE()
476 …auto recurrent_to_input_weights = create_tensor<CLTensor>(recurrent_weights_shape, DataType::QASY… in TEST_SUITE() local
497 …&recurrent_to_input_weights, &recurrent_to_forget_weights, &recurrent_to_cell_weights, &recurrent_… in TEST_SUITE()
[all …]
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayerQuantized.cpp | 67 …const ICLTensor *recurrent_to_input_weights, const ICLTensor *recurrent_to_forget_weights, const I… in configure() argument
72 …nput_to_forget_weights, input_to_cell_weights, input_to_output_weights, recurrent_to_input_weights, in configure()
79 …const ICLTensor *recurrent_to_input_weights, const ICLTensor *recurrent_to_forget_weights, const I… in configure() argument
85 …recurrent_to_input_weights, recurrent_to_forget_weights, recurrent_to_cell_weights, recurrent_to_o… in configure()
90 …recurrent_to_input_weights->info(), recurrent_to_forget_weights->info(), recurrent_to_cell_weights… in configure()
106 _recurrent_to_input_weights = recurrent_to_input_weights; in configure()
123 recurrent_weights_vector.emplace_back(recurrent_to_input_weights); in configure()
277 …const ITensorInfo *recurrent_to_input_weights, const ITensorInfo *recurrent_to_forget_weights, con… in validate() argument
282 …nput_to_forget_weights, input_to_cell_weights, input_to_output_weights, recurrent_to_input_weights, in validate()
305 …TE_RETURN_ERROR_ON_MISMATCHING_SHAPES(&recurrent_weights_info, recurrent_to_input_weights, recurre… in validate()
[all …]
|
| D | CLQLSTMLayer.cpp | 252 _recurrent_to_input_weights = lstm_params.recurrent_to_input_weights(); in configure()
282 …urrent_to_input_weights.configure(compile_context, lstm_params.recurrent_to_input_weights(), &_rec… in configure()
685 …N_ERROR(CLGEMMLowpMatrixAReductionKernel::validate(lstm_params.recurrent_to_input_weights(), &eff_… in validate()
720 …ARM_COMPUTE_RETURN_ON_ERROR(CLTranspose::validate(lstm_params.recurrent_to_input_weights(), &recur… in validate()
802 …R_ON_NULLPTR(lstm_params.input_to_input_weights(), lstm_params.recurrent_to_input_weights(), lstm_… in validate()
803 …to_forget_weights, lstm_params.input_to_input_weights(), lstm_params.recurrent_to_input_weights()); in validate()
805 …ERROR_ON_MISMATCHING_SHAPES(recurrent_to_forget_weights, lstm_params.recurrent_to_input_weights()); in validate()
814 …const float recurrent_to_input_scale = lstm_params.recurrent_to_input_weights()->quantization_info… in validate()
|
| D | CLLSTMLayer.cpp | 206 lstm_weights.emplace_back(lstm_params.recurrent_to_input_weights()); in configure()
527 lstm_params.recurrent_to_input_weights(), in validate()
530 ARM_COMPUTE_RETURN_ERROR_ON(lstm_params.recurrent_to_input_weights()->num_dimensions() > 2); in validate()
535 lstm_weights.emplace_back(lstm_params.recurrent_to_input_weights()); in validate()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.cpp | 71 …const ITensor *recurrent_to_input_weights, const ITensor *recurrent_to_forget_weights, const ITens… in configure() argument
77 …recurrent_to_input_weights, recurrent_to_forget_weights, recurrent_to_cell_weights, recurrent_to_o… in configure()
82 …recurrent_to_input_weights->info(), recurrent_to_forget_weights->info(), recurrent_to_cell_weights… in configure()
98 _recurrent_to_input_weights = recurrent_to_input_weights; in configure()
109 …std::vector<const ITensor *> recurrent_weights_vector{ recurrent_to_input_weights, recurrent_to_fo… in configure()
249 …const ITensorInfo *recurrent_to_input_weights, const ITensorInfo *recurrent_to_forget_weights, con… in validate() argument
254 …nput_to_forget_weights, input_to_cell_weights, input_to_output_weights, recurrent_to_input_weights, in validate()
276 …TE_RETURN_ERROR_ON_MISMATCHING_SHAPES(&recurrent_weights_info, recurrent_to_input_weights, recurre… in validate()
283 …ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(recurrent_to_input_weights, recurrent_to_forget… in validate()
290 …ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_QUANTIZATION_INFO(recurrent_to_input_weights, recurrent_to… in validate()
[all …]
|
| D | NEQLSTMLayer.cpp | 227 _recurrent_to_input_weights = lstm_params.recurrent_to_input_weights(); in configure()
268 …_transpose_recurrent_to_input_weights.configure(lstm_params.recurrent_to_input_weights(), &_recurr… in configure()
665 …N_ERROR(NEGEMMLowpMatrixAReductionKernel::validate(lstm_params.recurrent_to_input_weights(), &eff_… in validate()
699 …ARM_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(lstm_params.recurrent_to_input_weights(), &recur… in validate()
781 …R_ON_NULLPTR(lstm_params.input_to_input_weights(), lstm_params.recurrent_to_input_weights(), lstm_… in validate()
782 …to_forget_weights, lstm_params.input_to_input_weights(), lstm_params.recurrent_to_input_weights()); in validate()
784 …ERROR_ON_MISMATCHING_SHAPES(recurrent_to_forget_weights, lstm_params.recurrent_to_input_weights()); in validate()
793 …const float recurrent_to_input_scale = lstm_params.recurrent_to_input_weights()->quantization_info… in validate()
|
| D | NELSTMLayer.cpp | 181 lstm_weights.emplace_back(lstm_params.recurrent_to_input_weights()); in configure()
495 lstm_params.recurrent_to_input_weights(), in validate()
498 ARM_COMPUTE_RETURN_ERROR_ON(lstm_params.recurrent_to_input_weights()->num_dimensions() > 2); in validate()
503 lstm_weights.emplace_back(lstm_params.recurrent_to_input_weights()); in validate()
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayerQuantized.h | 96 …const ICLTensor *recurrent_to_input_weights, const ICLTensor *recurrent_to_forget_weights, const I…
123 …const ICLTensor *recurrent_to_input_weights, const ICLTensor *recurrent_to_forget_weights, const I…
152 …const ITensorInfo *recurrent_to_input_weights, const ITensorInfo *recurrent_to_forget_weights, con…
|
| /external/ComputeLibrary/arm_compute/runtime/common/ |
| D | LSTMParams.h | 84 …ams &set_cifg_params(const T *input_to_input_weights, const T *recurrent_to_input_weights, T *cell… in set_cifg_params() argument
87 _recurrent_to_input_weights = recurrent_to_input_weights; in set_cifg_params()
202 const T *recurrent_to_input_weights() const in recurrent_to_input_weights() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.h | 103 const TfLiteTensor* recurrent_to_input_weights,
135 const TfLiteTensor* recurrent_to_input_weights,
177 const TfLiteTensor* recurrent_to_input_weights,
203 const TfLiteTensor* recurrent_to_input_weights,
|
| D | lstm.cc | 196 const TfLiteTensor* recurrent_to_input_weights = in PopulateQuantizedLstmParams8x8_16() local
307 recurrent_to_input_weight_scale = recurrent_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
484 const TfLiteTensor* recurrent_to_input_weights = in PopulateQuantizedLstmParams8x8_8() local
612 recurrent_to_input_weight_ptr = recurrent_to_input_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
615 recurrent_to_input_weight_scale = recurrent_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
896 const TfLiteTensor* recurrent_to_input_weights = in CheckInputTensorDimensions() local
898 if (recurrent_to_input_weights != nullptr) { in CheckInputTensorDimensions()
899 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
900 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckInputTensorDimensions()
902 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckInputTensorDimensions()
[all …]
|
| D | unidirectional_sequence_lstm.cc | 112 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
229 recurrent_to_input_weight_scale = recurrent_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
453 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
455 if (recurrent_to_input_weights != nullptr) { in CheckInputTensorDimensions()
456 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
457 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckInputTensorDimensions()
459 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckInputTensorDimensions()
488 (recurrent_to_input_weights != nullptr)) || in CheckInputTensorDimensions()
490 (recurrent_to_input_weights == nullptr)); in CheckInputTensorDimensions()
742 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in PopulatePrecomputedZPTimesWeightsWithBias() local
[all …]
|
| D | lstm_eval.cc | 1692 const TfLiteTensor* recurrent_to_input_weights, in EvalFloat() argument
1779 GetTensorData<float>(recurrent_to_input_weights), in EvalFloat()
1840 GetTensorData<float>(recurrent_to_input_weights), in EvalFloat()
1877 const TfLiteTensor* recurrent_to_input_weights, in EvalHybrid() argument
1999 GetTensorData<int8_t>(recurrent_to_input_weights), in EvalHybrid()
2001 GetTensorScale(recurrent_to_input_weights), in EvalHybrid()
2097 GetTensorData<int8_t>(recurrent_to_input_weights), in EvalHybrid()
2099 GetTensorScale(recurrent_to_input_weights), in EvalHybrid()
2155 const TfLiteTensor* recurrent_to_input_weights, in EvalInteger8x8_16() argument
2218 GetTensorData<int8_t>(recurrent_to_input_weights), in EvalInteger8x8_16()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 237 const TfLiteTensor* recurrent_to_input_weights = in CheckLstmTensorDimensionsAndTypes() local
239 if (recurrent_to_input_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
240 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
241 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckLstmTensorDimensionsAndTypes()
243 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckLstmTensorDimensionsAndTypes()
245 TF_LITE_ENSURE_TYPES_EQ(context, recurrent_to_input_weights->type, in CheckLstmTensorDimensionsAndTypes()
276 (recurrent_to_input_weights != nullptr)) || in CheckLstmTensorDimensionsAndTypes()
278 (recurrent_to_input_weights == nullptr)); in CheckLstmTensorDimensionsAndTypes()
|
| D | lstm_test.cc | 1628 const std::vector<float> recurrent_to_input_weights = { in TEST() local
1713 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights); in TEST()
1790 const std::vector<float> recurrent_to_input_weights = { in TEST() local
1882 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights); in TEST()
1963 const std::vector<float> recurrent_to_input_weights = { in TEST() local
2140 const std::vector<float>& recurrent_to_input_weights, in HybridSparseLSTMOpModel() argument
2171 recurrent_weights_td, recurrent_to_input_weights, true); in HybridSparseLSTMOpModel()
|
| D | unidirectional_sequence_lstm_test.cc | 3064 const std::vector<float> recurrent_to_input_weights = { in TEST() local
3149 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights); in TEST()
3222 const std::vector<float> recurrent_to_input_weights = { in TEST() local
3314 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights); in TEST()
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | InfoHelpers.h | 87 …R_ON_NULLPTR(lstm_params.input_to_input_weights(), lstm_params.recurrent_to_input_weights(), lstm_… in build_lstm_params_tensor_info()
90 …rams(lstm_params.input_to_input_weights()->info(), lstm_params.recurrent_to_input_weights()->info(… in build_lstm_params_tensor_info()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.h | 100 …const ITensor *recurrent_to_input_weights, const ITensor *recurrent_to_forget_weights, const ITens…
129 …const ITensorInfo *recurrent_to_input_weights, const ITensorInfo *recurrent_to_forget_weights, con…
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 345 const TfLiteTensor* recurrent_to_input_weights, in EvalFloat() argument
435 GetTensorData<float>(recurrent_to_input_weights), in EvalFloat()
498 GetTensorData<float>(recurrent_to_input_weights), in EvalFloat()
560 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in lstm_eval() local
626 recurrent_to_input_weights, recurrent_to_forget_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 269 const TfLiteTensor* recurrent_to_input_weights, in EvalCalibration() argument
358 GetTensorData<float>(recurrent_to_input_weights), in EvalCalibration()
420 GetTensorData<float>(recurrent_to_input_weights), in EvalCalibration()
493 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in lstm_eval() local
611 recurrent_to_input_weights, recurrent_to_forget_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 44 std::initializer_list<float> recurrent_to_input_weights, in LSTMOpModel() argument
80 recurrent_to_input_weights); in LSTMOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 3836 TypeIsPred<"recurrent_to_input_weights", NoneType>,
3842 TypeIsPred<"recurrent_to_input_weights", NoneType>,
3903 TFL_OperandIsNoneOrHasRank<5, 2>, // recurrent_to_input_weights
3946 TFL_TensorOfOrNone<[F32, QI8]>:$recurrent_to_input_weights,
4033 TFL_OperandIsNoneOrHasRank<5, 2>, // recurrent_to_input_weights
4072 TFL_TensorOfOrNone<[F32, QI8]>:$recurrent_to_input_weights,
4354 TFL_TensorOf<[F32, QI8]>:$recurrent_to_input_weights,
|