Home
last modified time | relevance | path

Searched refs:projection_weights (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h121 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias,
157 const TfLiteTensor* projection_weights,
190 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias,
216 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias,
Dlstm_eval.cc265 const float* projection_weights, in CalculateLstmOutputFloat() argument
274 const bool use_projection = (projection_weights != nullptr); in CalculateLstmOutputFloat()
285 projection_weights, n_output, n_cell, scratch, n_batch, output_state); in CalculateLstmOutputFloat()
435 const int8_t* projection_weights, const uint8_t* projection_weights_ledger, in CalculateLstmOutputHybrid() argument
446 const bool use_projection = (projection_weights != nullptr); in CalculateLstmOutputHybrid()
467 projection_weights, projection_weights_ledger, n_output, n_cell, in CalculateLstmOutputHybrid()
471 projection_weights, n_output, n_cell, scratch1, in CalculateLstmOutputHybrid()
620 const int8_t* projection_weights, int32_t proj_scale_a, in CalculateLstmOutputInteger8x8_16() argument
631 const bool use_projection = (projection_weights != nullptr); in CalculateLstmOutputInteger8x8_16()
637 scratch1, projection_bias, projection_weights, proj_scale_a, in CalculateLstmOutputInteger8x8_16()
[all …]
Dunidirectional_sequence_lstm.cc142 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
153 const bool use_projection = (projection_weights != nullptr); in PopulateQuantizedLstmParams8x8_16()
243 projection_weight_scale = projection_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
578 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
580 if (projection_weights != nullptr) { in CheckInputTensorDimensions()
581 TF_LITE_ENSURE_EQ(context, projection_weights->dims->size, 2); in CheckInputTensorDimensions()
582 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[0], n_output); in CheckInputTensorDimensions()
583 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[1], n_cell); in CheckInputTensorDimensions()
604 ((projection_weights != nullptr) || (projection_bias == nullptr)); in CheckInputTensorDimensions()
753 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulatePrecomputedZPTimesWeightsWithBias() local
[all …]
Dlstm.cc227 const TfLiteTensor* projection_weights = in PopulateQuantizedLstmParams8x8_16() local
238 const bool use_projection = (projection_weights != nullptr); in PopulateQuantizedLstmParams8x8_16()
328 projection_weight_scale = projection_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
527 const TfLiteTensor* projection_weights = in PopulateQuantizedLstmParams8x8_8() local
543 const bool use_projection = (projection_weights != nullptr); in PopulateQuantizedLstmParams8x8_8()
643 projection_weight_ptr = projection_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
644 projection_weight_scale = projection_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
1028 const TfLiteTensor* projection_weights = in CheckInputTensorDimensions() local
1030 if (projection_weights != nullptr) { in CheckInputTensorDimensions()
1031 TF_LITE_ENSURE_EQ(context, projection_weights->dims->size, 2); in CheckInputTensorDimensions()
[all …]
Dbidirectional_sequence_lstm.cc353 const TfLiteTensor* projection_weights = in CheckLstmTensorDimensionsAndTypes() local
355 if (projection_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
356 TF_LITE_ENSURE_EQ(context, projection_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
357 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[0], n_output); in CheckLstmTensorDimensionsAndTypes()
358 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[1], n_cell); in CheckLstmTensorDimensionsAndTypes()
359 TF_LITE_ENSURE_TYPES_EQ(context, projection_weights->type, in CheckLstmTensorDimensionsAndTypes()
377 ((projection_weights != nullptr) || (projection_bias == nullptr)); in CheckLstmTensorDimensionsAndTypes()
Dlstm_test.cc1647 const std::vector<float> projection_weights = { in TEST() local
1718 lstm.SetProjectionWeights(projection_weights); in TEST()
1815 const std::vector<float> projection_weights = { in TEST() local
1891 lstm.SetProjectionWeights(projection_weights); in TEST()
1982 const std::vector<float> projection_weights = { in TEST() local
2057 lstm.SetProjectionWeights(projection_weights); in TEST()
Dunidirectional_sequence_lstm_test.cc3083 const std::vector<float> projection_weights = { in TEST() local
3154 lstm.SetProjectionWeights(projection_weights); in TEST()
3247 const std::vector<float> projection_weights = { in TEST() local
3323 lstm.SetProjectionWeights(projection_weights); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc126 const float* projection_weights, const float* projection_bias, in CalculateLstmOutputCalibration() argument
137 const bool use_projection = (projection_weights != nullptr); in CalculateLstmOutputCalibration()
148 projection_weights, n_output, n_cell, scratch, n_batch, output_state); in CalculateLstmOutputCalibration()
286 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalCalibration() argument
372 GetTensorData<float>(projection_weights), in EvalCalibration()
434 GetTensorData<float>(projection_weights), in EvalCalibration()
551 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in lstm_eval() local
621 projection_weights, projection_bias, &lstm_params, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc354 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalFloat() argument
441 GetTensorData<float>(projection_weights), in EvalFloat()
503 GetTensorData<float>(projection_weights), in EvalFloat()
586 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in lstm_eval() local
624 forget_gate_bias, cell_bias, output_gate_bias, projection_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc55 std::initializer_list<float> projection_weights, in LSTMOpModel() argument
112 AddConstInput({weight_type, {n_output, n_cell}}, projection_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc189 loc, int8, hidden_state.output(), lstm.projection_weights(), in LoadForLSTMOp()
Dprepare_quantize_helper.h99 !op.projection_weights().getType().template isa<NoneType>(); in GetLstmProperty()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3721 And<[TypeIsPred<"projection_weights", NoneType>,
3723 Neg<TypeIsPred<"projection_weights", NoneType>>]>>;
3807 TFL_OperandIsNoneOrHasRank<16, 2>, // projection_weights
3860 TFL_TensorOfOrNone<[F32, QI8]>:$projection_weights,
3939 TFL_OperandIsNoneOrHasRank<16, 2>, // projection_weights
3984 TFL_TensorOfOrNone<[F32, QI8]>:$projection_weights,