Home
last modified time | relevance | path

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

12

/external/ComputeLibrary/tests/validation/NEON/
DLSTMLayerQuantized.cpp103 …auto input_to_output_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE() local
124 …input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to_output_weights, in TEST_SUITE()
132 input_to_output_weights.allocator()->allocate(); in TEST_SUITE()
160 fill_tensor(input_to_output_weights, std::vector<uint8_t> { 203, 244, in TEST_SUITE()
242 …auto input_to_output_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_CASE() local
263 …input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to_output_weights, in TEST_CASE()
271 input_to_output_weights.allocator()->allocate(); in TEST_CASE()
311 …fill_tensor(input_to_output_weights, std::vector<uint8_t> { 205, 7, 95, 104, 252, 143, 226, 73, in TEST_CASE()
482 …auto input_to_output_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE() local
503 …input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to_output_weights, in TEST_SUITE()
[all …]
/external/ComputeLibrary/tests/validation/CL/
DLSTMLayerQuantized.cpp96 …auto input_to_output_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE() local
117 …input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to_output_weights, in TEST_SUITE()
125 input_to_output_weights.allocator()->allocate(); in TEST_SUITE()
153 fill_tensor(input_to_output_weights, std::vector<uint8_t> { 203, 244, in TEST_SUITE()
235 …auto input_to_output_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_CASE() local
256 …input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to_output_weights, in TEST_CASE()
264 input_to_output_weights.allocator()->allocate(); in TEST_CASE()
304 …fill_tensor(input_to_output_weights, std::vector<uint8_t> { 205, 7, 95, 104, 252, 143, 226, 73, in TEST_CASE()
475 …auto input_to_output_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE() local
496 …input_to_input_weights, &input_to_forget_weights, &input_to_cell_weights, &input_to_output_weights, in TEST_SUITE()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc106 const TfLiteTensor* input_to_output_weights; in PopulateQuantizedLstmParams8x8_16() local
110 &input_to_output_weights)); in PopulateQuantizedLstmParams8x8_16()
256 input_to_output_weight_scale = input_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
736 const TfLiteTensor* input_to_output_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
740 &input_to_output_weights)); in PopulatePrecomputedZPTimesWeightsWithBias()
822 context, input_zero_point, input_to_output_weights, output_gate_bias, in PopulatePrecomputedZPTimesWeightsWithBias()
897 const TfLiteTensor* input_to_output_weights; in Prepare() local
901 &input_to_output_weights)); in Prepare()
902 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare()
903 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2); in Prepare()
[all …]
Dlstm.cc191 const TfLiteTensor* input_to_output_weights; in PopulateQuantizedLstmParams8x8_16() local
194 &input_to_output_weights)); in PopulateQuantizedLstmParams8x8_16()
334 input_to_output_weight_scale = input_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
479 const TfLiteTensor* input_to_output_weights; in PopulateQuantizedLstmParams8x8_8() local
482 &input_to_output_weights)); in PopulateQuantizedLstmParams8x8_8()
655 input_to_output_weight_ptr = input_to_output_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
656 input_to_output_weight_scale = input_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
1174 const TfLiteTensor* input_to_output_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
1177 &input_to_output_weights)); in PopulatePrecomputedZPTimesWeightsWithBias()
1249 context, input_zero_point, input_to_output_weights, output_gate_bias, in PopulatePrecomputedZPTimesWeightsWithBias()
[all …]
Dlstm_eval.h102 const TfLiteTensor* input_to_output_weights,
133 const TfLiteTensor* input_to_output_weights,
176 const TfLiteTensor* input_to_output_weights,
202 const TfLiteTensor* input_to_output_weights,
Dlstm_eval.cc1691 const TfLiteTensor* input_to_output_weights, in EvalFloat() argument
1728 const int n_cell = input_to_output_weights->dims->data[0]; in EvalFloat()
1774 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
1835 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
1875 const TfLiteTensor* input_to_output_weights, in EvalHybrid() argument
1926 const int n_cell = input_to_output_weights->dims->data[0]; in EvalHybrid()
1988 GetTensorData<int8_t>(input_to_output_weights), in EvalHybrid()
1990 GetTensorScale(input_to_output_weights), aux_input_ptr, in EvalHybrid()
2086 GetTensorData<int8_t>(input_to_output_weights), in EvalHybrid()
2088 GetTensorScale(input_to_output_weights), aux_input_ptr, in EvalHybrid()
[all …]
Dlstm_test.cc1616 const std::vector<float> input_to_output_weights = { in TEST() local
1706 lstm.SetInputToOutputWeights(input_to_output_weights); in TEST()
1778 const std::vector<float> input_to_output_weights = { in TEST() local
1875 lstm.SetInputToOutputWeights(input_to_output_weights); in TEST()
1951 const std::vector<float> input_to_output_weights = { in TEST() local
2045 lstm.SetInputToOutputWeights(input_to_output_weights); in TEST()
2138 const std::vector<float>& input_to_output_weights, in HybridSparseLSTMOpModel() argument
2165 AddConstSparseInput(input_weights_td, input_to_output_weights, true); in HybridSparseLSTMOpModel()
Dbidirectional_sequence_lstm.cc227 const TfLiteTensor* input_to_output_weights; in CheckLstmTensorDimensionsAndTypes() local
230 &input_to_output_weights)); in CheckLstmTensorDimensionsAndTypes()
231 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
232 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
233 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
234 TF_LITE_ENSURE_TYPES_EQ(context, input_to_output_weights->type, in CheckLstmTensorDimensionsAndTypes()
Dunidirectional_sequence_lstm_test.cc3052 const std::vector<float> input_to_output_weights = { in TEST() local
3142 lstm.SetInputToOutputWeights(input_to_output_weights); in TEST()
3210 const std::vector<float> input_to_output_weights = { in TEST() local
3307 lstm.SetInputToOutputWeights(input_to_output_weights); in TEST()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayerQuantized.cpp66 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights, in configure() argument
72 …o_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, recurren… in configure()
78 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights, in configure() argument
84 …t, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
89 input_to_output_weights->info(), in configure()
105 _input_to_output_weights = input_to_output_weights; in configure()
120 inputs_weights_vector.emplace_back(input_to_output_weights); in configure()
276 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights, in validate() argument
282 …o_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, recurren… in validate()
304 …, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights); in validate()
[all …]
DCLQLSTMLayer.cpp165 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights, in configure() argument
172 …_compile_context(), input, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
178 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights, in configure() argument
185 …TE_ERROR_ON_NULLPTR(input, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
195 …), input_to_forget_weights->info(), input_to_cell_weights->info(), input_to_output_weights->info(), in configure()
202 const int num_units = input_to_output_weights->info()->dimension(1); in configure()
212 _input_to_output_weights = input_to_output_weights; in configure()
261 …_input_to_output_reduction->configure(compile_context, input_to_output_weights, &_input_to_output_… in configure()
275 …_transpose_input_to_output_weights.configure(compile_context, input_to_output_weights, &_input_to_… in configure()
455 …const float input_to_output_scale = input_to_output_weights->info()->quantization_info().unif… in configure()
[all …]
DCLLSTMLayer.cpp74 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights, in configure() argument
81 …ontext(), input, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, recurren… in configure()
87 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights, in configure() argument
95 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
109 … input_to_cell_weights->info(), input_to_output_weights->info(), in configure()
307 in_out_weights.emplace_back(input_to_output_weights); in configure()
409 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights, in validate() argument
417 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in validate()
426 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in validate()
436 ARM_COMPUTE_RETURN_ERROR_ON(input_to_output_weights->num_dimensions() > 2); in validate()
[all …]
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayerQuantized.cpp70 …ut_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input_to_output_weights, in configure() argument
76 …t, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
81 input_to_output_weights->info(), in configure()
97 _input_to_output_weights = input_to_output_weights; in configure()
108 … input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights }; in configure()
248 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights, in validate() argument
254 …o_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, recurren… in validate()
275 …, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights); in validate()
282 …, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights); in validate()
289 …, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights); in validate()
[all …]
DNEQLSTMLayer.cpp154 …ut_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input_to_output_weights, in configure() argument
161 …TE_ERROR_ON_NULLPTR(input, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
170 …), input_to_forget_weights->info(), input_to_cell_weights->info(), input_to_output_weights->info(), in configure()
177 const int num_units = input_to_output_weights->info()->dimension(1); in configure()
187 _input_to_output_weights = input_to_output_weights; in configure()
246 …_input_to_output_reduction->configure(input_to_output_weights, &_input_to_output_eff_bias, GEMMLow… in configure()
261 …_transpose_input_to_output_weights.configure(input_to_output_weights, &_input_to_output_weights_tr… in configure()
438 …const float input_to_output_scale = input_to_output_weights->info()->quantization_info().unif… in configure()
581 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights, in validate() argument
588 …N_NULLPTR(input, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, recurren… in validate()
[all …]
DNELSTMLayer.cpp67 …ut_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input_to_output_weights, in configure() argument
75 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in configure()
89 … input_to_cell_weights->info(), input_to_output_weights->info(), in configure()
279 in_out_weights.emplace_back(input_to_output_weights); in configure()
378 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights, in validate() argument
386 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in validate()
395 … input_to_forget_weights, input_to_cell_weights, input_to_output_weights, in validate()
405 ARM_COMPUTE_RETURN_ERROR_ON(input_to_output_weights->num_dimensions() > 2); in validate()
422 const unsigned int num_cells = input_to_output_weights->dimension(1); in validate()
553 in_out_weights.emplace_back(input_to_output_weights); in validate()
[all …]
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLLSTMLayerQuantized.h95 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights,
122 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights,
151 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights,
DCLLSTMLayer.h109 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights,
155 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights,
203 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights,
DCLQLSTMLayer.h117 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights,
167 …o_forget_weights, const ICLTensor *input_to_cell_weights, const ICLTensor *input_to_output_weights,
217 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights,
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc268 const TfLiteTensor* input_to_output_weights, in EvalCalibration() argument
307 const int n_cell = input_to_output_weights->dims->data[0]; in EvalCalibration()
353 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalCalibration()
415 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalCalibration()
486 const TfLiteTensor* input_to_output_weights; in lstm_eval() local
491 &input_to_output_weights)); in lstm_eval()
606 switch (input_to_output_weights->type) { 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.cc344 const TfLiteTensor* input_to_output_weights, in EvalFloat() argument
384 const int n_cell = input_to_output_weights->dims->data[0]; in EvalFloat()
430 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
493 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
557 const TfLiteTensor* input_to_output_weights = GetInput( in lstm_eval() local
621 switch (input_to_output_weights->type) { in lstm_eval()
625 input_to_cell_weights, input_to_output_weights, in lstm_eval()
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNELSTMLayerQuantized.h99 …ut_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input_to_output_weights,
128 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights,
DNELSTMLayer.h99 …ut_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input_to_output_weights,
146 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights,
DNEQLSTMLayer.h120 …ut_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input_to_output_weights,
170 …rget_weights, const ITensorInfo *input_to_cell_weights, const ITensorInfo *input_to_output_weights,
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc43 std::initializer_list<float> input_to_output_weights, in LSTMOpModel() argument
74 AddConstInput({weight_type, {n_cell, n_input}}, input_to_output_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2332 const auto input_to_output_weights = in Verify() local
2333 op.input_to_output_weights().getType().cast<ShapedType>(); in Verify()
2337 input_to_output_weights.hasStaticShape() && in Verify()
2340 const int n_cell = input_to_output_weights.getDimSize(0); in Verify()
2350 (input_to_output_weights.getDimSize(1) != n_input) || in Verify()
2353 (input_to_output_weights.getRank() != 2)) { in Verify()

12