Home
last modified time | relevance | path

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

12

/external/ComputeLibrary/tests/validation/CL/
DLSTMLayerQuantized.cpp102 auto forget_gate_bias = create_tensor<CLTensor>(bias_shape, DataType::S32); in TEST_SUITE() local
119 …&input_gate_bias, &forget_gate_bias, &cell_gate_bias, &output_gate_bias, &cell_state, &output_stat… in TEST_SUITE()
131 forget_gate_bias.allocator()->allocate(); in TEST_SUITE()
179 fill_tensor(forget_gate_bias, std::vector<int> { -23428, 126970, 116806, 46307 }); in TEST_SUITE()
241 auto forget_gate_bias = create_tensor<CLTensor>(bias_shape, DataType::S32); in TEST_CASE() local
258 …&input_gate_bias, &forget_gate_bias, &cell_gate_bias, &output_gate_bias, &cell_state, &output_stat… in TEST_CASE()
270 forget_gate_bias.allocator()->allocate(); in TEST_CASE()
352 …fill_tensor(forget_gate_bias, std::vector<int> { -128514, 8463, -57831, 116977, 106547, -281… in TEST_CASE()
481 auto forget_gate_bias = create_tensor<CLTensor>(bias_shape, DataType::S32); in TEST_SUITE() local
498 …&input_gate_bias, &forget_gate_bias, &cell_gate_bias, &output_gate_bias, &cell_state, &output_stat… in TEST_SUITE()
[all …]
/external/ComputeLibrary/tests/validation/NEON/
DLSTMLayerQuantized.cpp109 auto forget_gate_bias = create_tensor<Tensor>(bias_shape, DataType::S32); in TEST_SUITE() local
126 …&input_gate_bias, &forget_gate_bias, &cell_gate_bias, &output_gate_bias, &cell_state, &output_stat… in TEST_SUITE()
138 forget_gate_bias.allocator()->allocate(); in TEST_SUITE()
186 fill_tensor(forget_gate_bias, std::vector<int> { -23428, 126970, 116806, 46307 }); in TEST_SUITE()
248 auto forget_gate_bias = create_tensor<Tensor>(bias_shape, DataType::S32); in TEST_CASE() local
265 …&input_gate_bias, &forget_gate_bias, &cell_gate_bias, &output_gate_bias, &cell_state, &output_stat… in TEST_CASE()
277 forget_gate_bias.allocator()->allocate(); in TEST_CASE()
359 …fill_tensor(forget_gate_bias, std::vector<int> { -128514, 8463, -57831, 116977, 106547, -281… in TEST_CASE()
488 auto forget_gate_bias = create_tensor<Tensor>(bias_shape, DataType::S32); in TEST_SUITE() local
505 …&input_gate_bias, &forget_gate_bias, &cell_gate_bias, &output_gate_bias, &cell_state, &output_stat… in TEST_SUITE()
[all …]
/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayerQuantized.cpp63 …const ICLTensor *input_gate_bias, const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, c… in configure() argument
68 …rent_to_cell_weights, recurrent_to_output_weights, input_gate_bias, forget_gate_bias, cell_bias, o… in configure()
75 …const ICLTensor *input_gate_bias, const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, c… in configure() argument
81 …input_gate_bias, forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_state_in, ce… in configure()
84 …rent_to_cell_weights, recurrent_to_output_weights, input_gate_bias, forget_gate_bias, cell_bias, o… in configure()
90 …input_gate_bias->info(), forget_gate_bias->info(), cell_bias->info(), output_gate_bias->info(), ce… in configure()
110 _forget_gate_bias = forget_gate_bias; in configure()
153 bias_vector.emplace_back(forget_gate_bias); in configure()
283 …const ITensorInfo *input_gate_bias, const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_b… in validate() argument
288 …rent_to_cell_weights, recurrent_to_output_weights, input_gate_bias, forget_gate_bias, cell_bias, o… in validate()
[all …]
DCLLSTMLayer.cpp63 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias, in configure() argument
69 …recurrent_to_output_weights, forget_gate_bias, cell_bias, output_gate_bias, output_state_in, cell_… in configure()
76 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias, in configure() argument
84 forget_gate_bias, cell_bias, output_gate_bias, in configure()
89 …recurrent_to_output_weights, forget_gate_bias, cell_bias, output_gate_bias, output_state_in, cell_… in configure()
102 … forget_gate_bias->info(), cell_bias->info(), output_gate_bias->info(), in configure()
135 …et_gate_out2, &_forget_gate_out6, (_is_layer_norm_lstm) ? nullptr : forget_gate_bias, &_forget_gat… in configure()
168 …rget_gate_bias.configure(compile_context, &_forget_layer_norm_out1, forget_gate_bias, &_forget_lay… in configure()
403 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_… in validate() argument
411 forget_gate_bias, cell_bias, output_gate_bias, in validate()
[all …]
DCLQLSTMLayer.cpp162 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias, in configure() argument
168 …et_weights, recurrent_to_cell_weights, recurrent_to_output_weights, forget_gate_bias, cell_bias, o… in configure()
175 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias, in configure() argument
182 … forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_state_in, in configure()
187 … forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_state_in, in configure()
196 … forget_gate_bias->info(), cell_bias->info(), output_gate_bias->info(), in configure()
226 set_layer_norm_bias(forget_gate_bias, LayerNormGate::Forget); in configure()
606 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_… in validate() argument
612 …recurrent_to_output_weights, forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_… in validate()
633 ARM_COMPUTE_RETURN_ERROR_ON(forget_gate_bias->num_dimensions() != 1); in validate()
[all …]
/external/ComputeLibrary/tests/validation/fixtures/
DLSTMLayerFixture.h94 TensorType forget_gate_bias = create_tensor<TensorType>(cell_bias_shape, data_type); in compute_target() local
165 … &recurrent_to_cell_w, &recurrent_to_output_w, &forget_gate_bias, &cell_bias, &output_gate_bias, in compute_target()
177 ARM_COMPUTE_ASSERT(forget_gate_bias.info()->is_resizable()); in compute_target()
195 forget_gate_bias.allocator()->allocate(); in compute_target()
212 ARM_COMPUTE_ASSERT(!forget_gate_bias.info()->is_resizable()); in compute_target()
230 fill(AccessorType(forget_gate_bias), 7); in compute_target()
354 SimpleTensor<T> forget_gate_bias{ cell_bias_shape, data_type }; in compute_reference()
378 fill_custom_val(forget_gate_bias, 0.f, 7); in compute_reference()
384 fill(forget_gate_bias, 7); in compute_reference()
409 …forget = reference::fully_connected_layer(input, input_to_forget_w, forget_gate_bias, output_cell_… in compute_reference()
[all …]
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayer.cpp60 … const ITensor *forget_gate_bias, const ITensor *cell_bias, const ITensor *output_gate_bias, in configure() argument
68 forget_gate_bias, cell_bias, output_gate_bias, in configure()
74 forget_gate_bias, cell_bias, output_gate_bias, in configure()
89 … forget_gate_bias->info(), cell_bias->info(), output_gate_bias->info(), in configure()
119 …et_gate_out2, &_forget_gate_out6, (_is_layer_norm_lstm) ? nullptr : forget_gate_bias, &_forget_gat… in configure()
151 …_accum_forget_gate_bias.configure(&_forget_layer_norm_out1, forget_gate_bias, &_forget_layer_norm_… in configure()
378 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_… in validate() argument
386 forget_gate_bias, cell_bias, output_gate_bias, in validate()
395 … forget_gate_bias, cell_bias, output_gate_bias, in validate()
407 ARM_COMPUTE_RETURN_ERROR_ON(forget_gate_bias->num_dimensions() > 1); in validate()
[all …]
DNELSTMLayerQuantized.cpp63 …const ITensor *input_gate_bias, const ITensor *forget_gate_bias, const ITensor *cell_bias, const I… in configure() argument
69 …input_gate_bias, forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_state_in, ce… in configure()
74 …input_gate_bias->info(), forget_gate_bias->info(), cell_bias->info(), output_gate_bias->info(), ce… in configure()
78 …input_gate_bias, forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_state_in, ce… in configure()
98 _forget_gate_bias = forget_gate_bias; in configure()
124 …std::vector<const ITensor *> bias_vector{ input_gate_bias, forget_gate_bias, cell_bias, output_gat… in configure()
251 …const ITensorInfo *input_gate_bias, const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_b… in validate() argument
256 …rent_to_cell_weights, recurrent_to_output_weights, input_gate_bias, forget_gate_bias, cell_bias, o… in validate()
278 …ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(&bias_info, input_gate_bias, forget_gate_bias, cell… in validate()
285 …RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(&bias_info, input_gate_bias, forget_gate_bias, cell_bias, o… in validate()
[all …]
DNEQLSTMLayer.cpp217 … const ITensor *forget_gate_bias, const ITensor *cell_bias, const ITensor *output_gate_bias, in configure() argument
224 …forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_state_in, cell_state_out, out… in configure()
228 …forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_state_in, cell_state_out, out… in configure()
258 … forget_gate_bias->info(), cell_bias->info(), output_gate_bias->info(), in configure()
266 … forget_gate_bias->info(), cell_bias->info(), output_gate_bias->info(), in configure()
297 set_layer_norm_bias(forget_gate_bias, LayerNormGate::Forget); in configure()
678 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_… in validate() argument
684 …recurrent_to_output_weights, forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_… in validate()
714 ARM_COMPUTE_RETURN_ERROR_ON(forget_gate_bias->num_dimensions() != 1); in validate()
715 ARM_COMPUTE_RETURN_ERROR_ON(forget_gate_bias->dimension(0) != num_units); in validate()
[all …]
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLLSTMLayerQuantized.h105 …const ICLTensor *input_gate_bias, const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, c…
132 …const ICLTensor *input_gate_bias, const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, c…
161 …const ITensorInfo *input_gate_bias, const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_b…
DCLLSTMLayer.h126 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias,
172 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias,
220 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_…
DCLQLSTMLayer.h133 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias,
183 … const ICLTensor *forget_gate_bias, const ICLTensor *cell_bias, const ICLTensor *output_gate_bias,
233 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_…
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h119 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
155 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
188 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
214 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias,
Dunidirectional_sequence_lstm.cc550 const TfLiteTensor* forget_gate_bias; in CheckInputTensorDimensions() local
553 &forget_gate_bias)); in CheckInputTensorDimensions()
554 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->size, 1); in CheckInputTensorDimensions()
555 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->data[0], n_cell); in CheckInputTensorDimensions()
557 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteInt32); in CheckInputTensorDimensions()
559 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteFloat32); in CheckInputTensorDimensions()
784 const TfLiteTensor* forget_gate_bias = in PopulatePrecomputedZPTimesWeightsWithBias() local
791 context, input_zero_point, input_to_forget_weights, forget_gate_bias, in PopulatePrecomputedZPTimesWeightsWithBias()
1273 const TfLiteTensor* forget_gate_bias; in Eval() local
1276 &forget_gate_bias)); in Eval()
[all …]
Dlstm.cc518 const TfLiteTensor* forget_gate_bias; in PopulateQuantizedLstmParams8x8_8() local
520 &forget_gate_bias)); in PopulateQuantizedLstmParams8x8_8()
664 forget_gate_bias_ptr = forget_gate_bias->data.i32; in PopulateQuantizedLstmParams8x8_8()
996 const TfLiteTensor* forget_gate_bias; in CheckInputTensorDimensions() local
998 &forget_gate_bias)); in CheckInputTensorDimensions()
999 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->size, 1); in CheckInputTensorDimensions()
1000 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->data[0], n_cell); in CheckInputTensorDimensions()
1002 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteInt32); in CheckInputTensorDimensions()
1004 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteFloat32); in CheckInputTensorDimensions()
1216 const TfLiteTensor* forget_gate_bias = in PopulatePrecomputedZPTimesWeightsWithBias() local
[all …]
Dlstm_eval.cc1766 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalFloat() argument
1853 GetTensorData<float>(forget_gate_bias), in EvalFloat()
1915 GetTensorData<float>(forget_gate_bias), in EvalFloat()
1961 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalHybrid()
2086 GetTensorData<float>(forget_gate_bias), in EvalHybrid()
2184 GetTensorData<float>(forget_gate_bias), in EvalHybrid()
2230 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalInteger8x8_16() argument
2322 GetTensorData<int32_t>(forget_gate_bias), in EvalInteger8x8_16()
2421 GetTensorData<int32_t>(forget_gate_bias), in EvalInteger8x8_16()
2468 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalInteger8x8_8() argument
[all …]
Dlstm_test.cc1622 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
1710 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
1784 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
1879 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
1957 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
2049 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
Dbidirectional_sequence_lstm.cc330 const TfLiteTensor* forget_gate_bias; in CheckLstmTensorDimensionsAndTypes() local
333 GetInputSafe(context, node, forget_gate_bias_tensor, &forget_gate_bias)); in CheckLstmTensorDimensionsAndTypes()
334 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
335 TF_LITE_ENSURE_EQ(context, forget_gate_bias->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
336 TF_LITE_ENSURE_TYPES_EQ(context, forget_gate_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
Dunidirectional_sequence_lstm_test.cc2806 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
2894 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
2964 const std::vector<float> forget_gate_bias = {0.1, -0.3, -0.2, 0.1}; in TEST() local
3059 lstm.SetForgetGateBias(forget_gate_bias); in TEST()
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNELSTMLayerQuantized.h109 …const ITensor *input_gate_bias, const ITensor *forget_gate_bias, const ITensor *cell_bias, const I…
138 …const ITensorInfo *input_gate_bias, const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_b…
DNELSTMLayer.h110 … const ITensor *forget_gate_bias, const ITensor *cell_bias, const ITensor *output_gate_bias,
157 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_…
DNEQLSTMLayer.h136 … const ITensor *forget_gate_bias, const ITensor *cell_bias, const ITensor *output_gate_bias,
186 …const ITensorInfo *forget_gate_bias, const ITensorInfo *cell_bias, const ITensorInfo *output_gate_…
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc285 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalCalibration() argument
370 GetTensorData<float>(forget_gate_bias), in EvalCalibration()
432 GetTensorData<float>(forget_gate_bias), in EvalCalibration()
536 const TfLiteTensor* forget_gate_bias; in lstm_eval() local
540 &forget_gate_bias)); in lstm_eval()
621 forget_gate_bias, cell_gate_bias, output_gate_bias, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc361 const TfLiteTensor* input_gate_bias, const TfLiteTensor* forget_gate_bias, in EvalFloat() argument
447 GetTensorData<float>(forget_gate_bias), in EvalFloat()
510 GetTensorData<float>(forget_gate_bias), in EvalFloat()
591 const TfLiteTensor* forget_gate_bias = in lstm_eval() local
636 forget_gate_bias, cell_bias, output_gate_bias, projection_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc52 std::initializer_list<float> forget_gate_bias, in LSTMOpModel() argument
107 AddConstInput({TensorType_FLOAT32, {n_cell}}, forget_gate_bias); in LSTMOpModel()

12