Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h109 const TfLiteTensor* cell_to_output_weights,
145 const TfLiteTensor* cell_to_output_weights,
183 const TfLiteTensor* cell_to_output_weights,
209 const TfLiteTensor* cell_to_output_weights,
Dunidirectional_sequence_lstm.cc130 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
152 const bool use_peephole = (cell_to_output_weights != nullptr); in PopulateQuantizedLstmParams8x8_16()
230 cell_to_output_weight_scale = cell_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
506 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
508 if (cell_to_output_weights != nullptr) { in CheckInputTensorDimensions()
509 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->size, 1); in CheckInputTensorDimensions()
510 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
512 context, cell_to_output_weights->type, in CheckInputTensorDimensions()
521 (cell_to_output_weights != nullptr)) || in CheckInputTensorDimensions()
524 (cell_to_output_weights == nullptr)); in CheckInputTensorDimensions()
[all …]
Dlstm.cc215 const TfLiteTensor* cell_to_output_weights = in PopulateQuantizedLstmParams8x8_16() local
237 const bool use_peephole = (cell_to_output_weights != nullptr); in PopulateQuantizedLstmParams8x8_16()
315 cell_to_output_weight_scale = cell_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
503 const TfLiteTensor* cell_to_output_weights = in PopulateQuantizedLstmParams8x8_8() local
541 const bool use_peephole = (cell_to_output_weights != nullptr); in PopulateQuantizedLstmParams8x8_8()
624 cell_to_output_weight_ptr = cell_to_output_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
626 cell_to_output_weight_scale = cell_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
960 const TfLiteTensor* cell_to_output_weights = in CheckInputTensorDimensions() local
962 if (cell_to_output_weights) { in CheckInputTensorDimensions()
963 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->size, 1); in CheckInputTensorDimensions()
[all …]
Dlstm_eval.cc1698 const TfLiteTensor* cell_to_output_weights, in EvalFloat() argument
1785 GetTensorData<float>(cell_to_output_weights), in EvalFloat()
1846 GetTensorData<float>(cell_to_output_weights), in EvalFloat()
1887 const TfLiteTensor* cell_to_output_weights, in EvalHybrid() argument
2015 GetTensorData<int8_t>(cell_to_output_weights), in EvalHybrid()
2016 GetTensorScale(cell_to_output_weights), in EvalHybrid()
2113 GetTensorData<int8_t>(cell_to_output_weights), in EvalHybrid()
2114 GetTensorScale(cell_to_output_weights), in EvalHybrid()
2161 const TfLiteTensor* cell_to_output_weights, in EvalInteger8x8_16() argument
2236 GetTensorData<int16_t>(cell_to_output_weights), in EvalInteger8x8_16()
[all …]
Dbidirectional_sequence_lstm.cc299 const TfLiteTensor* cell_to_output_weights = in CheckLstmTensorDimensionsAndTypes() local
301 if (cell_to_output_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
302 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
303 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
304 TF_LITE_ENSURE_TYPES_EQ(context, cell_to_output_weights->type, in CheckLstmTensorDimensionsAndTypes()
313 (cell_to_output_weights != nullptr)) || in CheckLstmTensorDimensionsAndTypes()
316 (cell_to_output_weights == nullptr)); in CheckLstmTensorDimensionsAndTypes()
Dlstm_test.cc1806 const std::vector<float> cell_to_output_weights = {0.3, -0.1, 0.1, -0.3}; in TEST() local
1889 lstm.SetCellToOutputWeights(cell_to_output_weights); in TEST()
Dunidirectional_sequence_lstm_test.cc3238 const std::vector<float> cell_to_output_weights = {0.3, -0.1, 0.1, -0.3}; in TEST() local
3321 lstm.SetCellToOutputWeights(cell_to_output_weights); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc342 const TfLiteTensor* cell_to_output_weights, in EvalFloat() argument
432 GetTensorData<float>(cell_to_output_weights), in EvalFloat()
494 GetTensorData<float>(cell_to_output_weights), in EvalFloat()
561 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in lstm_eval() local
616 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc274 const TfLiteTensor* cell_to_output_weights, in EvalCalibration() argument
363 GetTensorData<float>(cell_to_output_weights), in EvalCalibration()
425 GetTensorData<float>(cell_to_output_weights), in EvalCalibration()
517 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in lstm_eval() local
612 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc50 std::initializer_list<float> cell_to_output_weights, in LSTMOpModel() argument
95 AddConstInput({weight_type, {n_cell}}, cell_to_output_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc182 {new_cell, lstm.cell_to_output_weights()}), in LoadForLSTMOp()
Dprepare_quantize_helper.h101 !op.cell_to_output_weights().getType().template isa<NoneType>(); in GetLstmProperty()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3803 TFL_OperandIsNoneOrHasRank<11, 1>, // cell_to_output_weights
3851 TFL_TensorOfOrNone<[F32, QI8, QI16]>:$cell_to_output_weights,
3934 TFL_OperandIsNoneOrHasRank<11, 1>, // cell_to_output_weights
3975 TFL_TensorOfOrNone<[F32, QI8]>:$cell_to_output_weights,