| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 102 const TfLiteTensor* input_to_output_weights; in PopulateQuantizedLstmParams8x8_16() local
106 &input_to_output_weights)); in PopulateQuantizedLstmParams8x8_16()
249 input_to_output_weight_scale = input_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
729 const TfLiteTensor* input_to_output_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
733 &input_to_output_weights)); in PopulatePrecomputedZPTimesWeightsWithBias()
813 context, input_zero_point, input_to_output_weights, output_gate_bias, in PopulatePrecomputedZPTimesWeightsWithBias()
888 const TfLiteTensor* input_to_output_weights; in Prepare() local
892 &input_to_output_weights)); in Prepare()
893 const int n_cell = input_to_output_weights->dims->data[0]; in Prepare()
894 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2); in Prepare()
[all …]
|
| D | lstm.cc | 191 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 …]
|
| D | lstm_eval.h | 102 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,
|
| D | lstm_eval.cc | 1691 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 …]
|
| D | lstm_test.cc | 1616 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()
|
| D | bidirectional_sequence_lstm.cc | 227 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()
|
| D | unidirectional_sequence_lstm_test.cc | 3052 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/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 267 const TfLiteTensor* input_to_output_weights, in EvalCalibration() argument
306 const int n_cell = input_to_output_weights->dims->data[0]; in EvalCalibration()
352 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalCalibration()
414 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalCalibration()
485 const TfLiteTensor* input_to_output_weights; in lstm_eval() local
490 &input_to_output_weights)); in lstm_eval()
605 switch (input_to_output_weights->type) { in lstm_eval()
609 input_to_cell_weights, input_to_output_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 335 const TfLiteTensor* input_to_output_weights, in EvalFloat() argument
375 const int n_cell = input_to_output_weights->dims->data[0]; in EvalFloat()
421 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
483 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
545 const TfLiteTensor* input_to_output_weights = GetInput( in lstm_eval() local
609 switch (input_to_output_weights->type) { in lstm_eval()
613 input_to_cell_weights, input_to_output_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 43 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/transforms/ |
| D | load_quantization_recipe.cc | 179 loc, lstm.input(), lstm.input_to_output_weights(), in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2020 const auto input_to_output_weights = in Verify() local
2021 op.input_to_output_weights().getType().cast<ShapedType>(); in Verify()
2025 input_to_output_weights.hasStaticShape() && in Verify()
2028 const int n_cell = input_to_output_weights.getDimSize(0); in Verify()
2038 (input_to_output_weights.getDimSize(1) != n_input) || in Verify()
2041 (input_to_output_weights.getRank() != 2)) { in Verify()
|
| D | tfl_ops.td | 3838 TFL_TensorOf<[F32, QI8]>:$input_to_output_weights,
3927 TFL_OperandHasRank<4, 2>, // input_to_output_weights
3962 TFL_TensorOf<[F32, QI8]>:$input_to_output_weights,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | ops.mlir | 843 // 'input_to_output_weights' input for lstm op has unmatched rank with `input`.
854 …efficient inputs of LSTM op don't match the dimension with input operand `input_to_output_weights`.
858 …more than 2 dimensions or don't match the dimension with input operand `input_to_output_weights`.}}
869 …more than 2 dimensions or don't match the dimension with input operand `input_to_output_weights`.}}
|