Home
last modified time | relevance | path

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

/external/ComputeLibrary/tests/validation/NEON/
DLSTMLayerQuantized.cpp102 …auto input_to_cell_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE() local
124 …igure(&input, &input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to… in TEST_SUITE()
131 input_to_cell_weights.allocator()->allocate(); in TEST_SUITE()
155 fill_tensor(input_to_cell_weights, std::vector<uint8_t> { 172, 101, in TEST_SUITE()
241 …auto input_to_cell_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_CASE() local
263 …igure(&input, &input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to… in TEST_CASE()
270 input_to_cell_weights.allocator()->allocate(); in TEST_CASE()
302 … fill_tensor(input_to_cell_weights, std::vector<uint8_t> { 93, 103, 226, 139, 185, 252, 129, 171, in TEST_CASE()
481 …auto input_to_cell_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE() local
503 …igure(&input, &input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to… in TEST_SUITE()
[all …]
/external/ComputeLibrary/tests/validation/CL/
DLSTMLayerQuantized.cpp95 …auto input_to_cell_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE() local
117 …igure(&input, &input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to… in TEST_SUITE()
124 input_to_cell_weights.allocator()->allocate(); in TEST_SUITE()
148 fill_tensor(input_to_cell_weights, std::vector<uint8_t> { 172, 101, in TEST_SUITE()
234 …auto input_to_cell_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_CASE() local
256 …igure(&input, &input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to… in TEST_CASE()
263 input_to_cell_weights.allocator()->allocate(); in TEST_CASE()
295 … fill_tensor(input_to_cell_weights, std::vector<uint8_t> { 93, 103, 226, 139, 185, 252, 129, 171, in TEST_CASE()
474 …auto input_to_cell_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE() local
496 …igure(&input, &input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to… in TEST_SUITE()
[all …]
/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayerQuantized.cpp61 …ights, const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICL… in configure() argument
67 …ontext(), input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in configure()
73 …ights, const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICL… in configure() argument
79 …N_NULLPTR(input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in configure()
83 …OG_PARAMS(input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in configure()
87 …(), input_to_input_weights->info(), input_to_forget_weights->info(), input_to_cell_weights->info(), in configure()
103 _input_to_cell_weights = input_to_cell_weights; in configure()
118 inputs_weights_vector.emplace_back(input_to_cell_weights); in configure()
281 …s, const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITe… in validate() argument
287 …N_NULLPTR(input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in validate()
[all …]
DCLLSTMLayer.cpp61 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor … in configure() argument
68 …ary::get().get_compile_context(), input, input_to_forget_weights, input_to_cell_weights, input_to_… in configure()
74 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor … in configure() argument
82 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
88 …ARM_COMPUTE_LOG_PARAMS(input, input_to_forget_weights, input_to_cell_weights, input_to_output_weig… in configure()
100 … input_to_cell_weights->info(), input_to_output_weights->info(), in configure()
252 …_fully_connected_cell_state.configure(compile_context, input, input_to_cell_weights, (_is_layer_no… in configure()
401 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso… in validate() argument
409 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in validate()
418 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in validate()
[all …]
DCLQLSTMLayer.cpp160 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor … in configure() argument
167 …ary::get().get_compile_context(), input, input_to_forget_weights, input_to_cell_weights, input_to_… in configure()
173 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor … in configure() argument
180 …ARM_COMPUTE_ERROR_ON_NULLPTR(input, input_to_forget_weights, input_to_cell_weights, input_to_outpu… in configure()
185 …ARM_COMPUTE_LOG_PARAMS(input, input_to_forget_weights, input_to_cell_weights, input_to_output_weig… in configure()
194 …TMLayer::validate(input->info(), input_to_forget_weights->info(), input_to_cell_weights->info(), i… in configure()
210 _input_to_cell_weights = input_to_cell_weights; in configure()
260 …_input_to_cell_reduction->configure(compile_context, input_to_cell_weights->info(), _input_to_cell… in configure()
277 …_transpose_input_to_cell_weights.configure(compile_context, input_to_cell_weights, &_input_to_cell… in configure()
353 …const float input_to_cell_scale = input_to_cell_weights->info()->quantization_info().uniform(… in configure()
[all …]
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLLSTMLayerQuantized.h103 …ights, const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICL…
130 …ights, const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICL…
159 …s, const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITe…
DCLLSTMLayer.h124 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor …
170 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor …
218 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso…
DCLQLSTMLayer.h131 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor …
181 …const ICLTensor *input_to_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor …
231 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso…
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayerQuantized.cpp61 …t_weights, const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITe… in configure() argument
67 …N_NULLPTR(input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in configure()
71 …(), input_to_input_weights->info(), input_to_forget_weights->info(), input_to_cell_weights->info(), in configure()
76 …OG_PARAMS(input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in configure()
91 _input_to_cell_weights = input_to_cell_weights; in configure()
103 …_weights_vector{ input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in configure()
249 …s, const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITe… in validate() argument
255 …N_NULLPTR(input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in validate()
276 …ut_weights_info, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in validate()
283 …hts_info, input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_… in validate()
[all …]
DNEQLSTMLayer.cpp215 …const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input… in configure() argument
222 …ARM_COMPUTE_ERROR_ON_NULLPTR(input, input_to_forget_weights, input_to_cell_weights, input_to_outpu… in configure()
226 …ARM_COMPUTE_LOG_PARAMS(input, input_to_forget_weights, input_to_cell_weights, input_to_output_weig… in configure()
235 …_input_to_cell_weights_transposed.info()->set_quantization_info(input_to_cell_weights->info()->qua… in configure()
256 …r::validate(input->info(), _input_to_forget_weights_symm8.info(), input_to_cell_weights->info(), i… in configure()
264 …TMLayer::validate(input->info(), input_to_forget_weights->info(), input_to_cell_weights->info(), i… in configure()
281 _input_to_cell_weights = input_to_cell_weights; in configure()
339 …_input_to_cell_reduction->configure(input_to_cell_weights->info(), _input_to_cell_eff_bias.info(),… in configure()
355 …_transpose_input_to_cell_weights.configure(input_to_cell_weights, &_input_to_cell_weights_transpos… in configure()
429 …const float input_to_cell_scale = input_to_cell_weights->info()->quantization_info().uniform(… in configure()
[all …]
DNELSTMLayer.cpp58 …const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input… in configure() argument
66 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
72 input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
87 … input_to_cell_weights->info(), input_to_output_weights->info(), in configure()
231 …_fully_connected_cell_state.configure(input, input_to_cell_weights, (_is_layer_norm_lstm) ? nullpt… in configure()
376 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso… in validate() argument
384 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in validate()
393 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in validate()
402 ARM_COMPUTE_RETURN_ERROR_ON(input_to_cell_weights->num_dimensions() > 2); in validate()
529 …ARM_COMPUTE_RETURN_ON_ERROR(NEFullyConnectedLayer::validate(input, input_to_cell_weights, (lstm_pa… in validate()
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h101 const TfLiteTensor* input_to_cell_weights,
131 const TfLiteTensor* input_to_cell_weights,
175 const TfLiteTensor* input_to_cell_weights,
201 const TfLiteTensor* input_to_cell_weights,
Dlstm.cc188 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_16() local
191 &input_to_cell_weights)); in PopulateQuantizedLstmParams8x8_16()
334 input_to_cell_weight_scale = input_to_cell_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
476 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_8() local
479 &input_to_cell_weights)); in PopulateQuantizedLstmParams8x8_8()
654 input_to_cell_weight_ptr = input_to_cell_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
655 input_to_cell_weight_scale = input_to_cell_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
887 const TfLiteTensor* input_to_cell_weights; in CheckInputTensorDimensions() local
890 &input_to_cell_weights)); in CheckInputTensorDimensions()
891 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
[all …]
Dunidirectional_sequence_lstm.cc103 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_16() local
106 &input_to_cell_weights)); in PopulateQuantizedLstmParams8x8_16()
256 input_to_cell_weight_scale = input_to_cell_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
446 const TfLiteTensor* input_to_cell_weights; in CheckInputTensorDimensions() local
449 &input_to_cell_weights)); in CheckInputTensorDimensions()
450 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
451 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
452 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
733 const TfLiteTensor* input_to_cell_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
736 &input_to_cell_weights)); in PopulatePrecomputedZPTimesWeightsWithBias()
[all …]
Dlstm_test.cc1612 const std::vector<float> input_to_cell_weights = { in TEST() local
1704 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
1774 const std::vector<float> input_to_cell_weights = { in TEST() local
1873 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
1947 const std::vector<float> input_to_cell_weights = { in TEST() local
2043 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
2137 const std::vector<float>& input_to_cell_weights, in HybridSparseLSTMOpModel() argument
2162 AddConstSparseInput(input_weights_td, input_to_cell_weights, true); in HybridSparseLSTMOpModel()
Dlstm_eval.cc1748 const TfLiteTensor* input_to_cell_weights, in EvalFloat() argument
1835 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
1897 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
1937 const TfLiteTensor* input_to_cell_weights, in EvalHybrid() argument
2049 GetTensorData<int8_t>(input_to_cell_weights), in EvalHybrid()
2051 GetTensorScale(input_to_cell_weights), in EvalHybrid()
2147 GetTensorData<int8_t>(input_to_cell_weights), in EvalHybrid()
2149 GetTensorScale(input_to_cell_weights), in EvalHybrid()
2217 const TfLiteTensor* input_to_cell_weights, in EvalInteger8x8_16() argument
2276 GetTensorData<int8_t>(input_to_cell_weights), in EvalInteger8x8_16()
[all …]
Dbidirectional_sequence_lstm.cc217 const TfLiteTensor* input_to_cell_weights; in CheckLstmTensorDimensionsAndTypes() local
220 &input_to_cell_weights)); in CheckLstmTensorDimensionsAndTypes()
221 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
222 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
223 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
224 TF_LITE_ENSURE_TYPES_EQ(context, input_to_cell_weights->type, in CheckLstmTensorDimensionsAndTypes()
Dunidirectional_sequence_lstm_test.cc2796 const std::vector<float> input_to_cell_weights = { in TEST() local
2888 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
2954 const std::vector<float> input_to_cell_weights = { in TEST() local
3053 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNELSTMLayerQuantized.h107 …t_weights, const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITe…
136 …s, const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITe…
DNELSTMLayer.h108 …const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input…
155 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso…
DNEQLSTMLayer.h134 …const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input…
184 …const ITensorInfo *input_to_forget_weights, const ITensorInfo *input_to_cell_weights, const ITenso…
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc267 const TfLiteTensor* input_to_cell_weights, in EvalCalibration() argument
352 GetTensorData<float>(input_to_cell_weights), in EvalCalibration()
414 GetTensorData<float>(input_to_cell_weights), in EvalCalibration()
481 const TfLiteTensor* input_to_cell_weights; in lstm_eval() local
485 &input_to_cell_weights)); in lstm_eval()
610 input_to_cell_weights, input_to_output_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc343 const TfLiteTensor* input_to_cell_weights, in EvalFloat() argument
429 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
492 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
555 const TfLiteTensor* input_to_cell_weights = GetInput( in lstm_eval() local
625 input_to_cell_weights, input_to_output_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc42 std::initializer_list<float> input_to_cell_weights, in LSTMOpModel() argument
73 AddConstInput({weight_type, {n_cell, n_input}}, input_to_cell_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td4170 TFL_OperandHasRank<3, 2>, // input_to_cell_weights
4212 TFL_TensorOf<[F32, QI8]>:$input_to_cell_weights,
4310 TFL_OperandHasRank<3, 2>, // input_to_cell_weights
4349 TFL_TensorOf<[F32, QI8]>:$input_to_cell_weights,