Home
last modified time | relevance | path

Searched refs:cell_to_input_weights (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h107 const TfLiteTensor* cell_to_input_weights,
143 const TfLiteTensor* cell_to_input_weights,
181 const TfLiteTensor* cell_to_input_weights,
207 const TfLiteTensor* cell_to_input_weights,
Dunidirectional_sequence_lstm.cc126 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
227 cell_to_input_weight_scale = cell_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
486 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
488 if (cell_to_input_weights != nullptr) { in CheckInputTensorDimensions()
489 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->size, 1); in CheckInputTensorDimensions()
490 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
492 context, cell_to_input_weights->type, in CheckInputTensorDimensions()
519 ((cell_to_input_weights != nullptr || use_cifg) && in CheckInputTensorDimensions()
522 ((cell_to_input_weights == nullptr) && in CheckInputTensorDimensions()
1251 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in Eval() local
[all …]
Dlstm.cc211 const TfLiteTensor* cell_to_input_weights = in PopulateQuantizedLstmParams8x8_16() local
312 cell_to_input_weight_scale = cell_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
499 const TfLiteTensor* cell_to_input_weights = in PopulateQuantizedLstmParams8x8_8() local
620 cell_to_input_weight_ptr = cell_to_input_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
621 cell_to_input_weight_scale = cell_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
940 const TfLiteTensor* cell_to_input_weights = in CheckInputTensorDimensions() local
942 if (cell_to_input_weights) { in CheckInputTensorDimensions()
943 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->size, 1); in CheckInputTensorDimensions()
944 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
946 context, cell_to_input_weights->type, in CheckInputTensorDimensions()
[all …]
Dlstm_eval.cc1696 const TfLiteTensor* cell_to_input_weights, in EvalFloat() argument
1783 GetTensorData<float>(cell_to_input_weights), in EvalFloat()
1844 GetTensorData<float>(cell_to_input_weights), in EvalFloat()
1885 const TfLiteTensor* cell_to_input_weights, in EvalHybrid() argument
2011 GetTensorData<int8_t>(cell_to_input_weights), in EvalHybrid()
2012 GetTensorScale(cell_to_input_weights), in EvalHybrid()
2109 GetTensorData<int8_t>(cell_to_input_weights), in EvalHybrid()
2110 GetTensorScale(cell_to_input_weights), in EvalHybrid()
2159 const TfLiteTensor* cell_to_input_weights, in EvalInteger8x8_16() argument
2230 GetTensorData<int16_t>(cell_to_input_weights), in EvalInteger8x8_16()
[all …]
Dbidirectional_sequence_lstm.cc281 const TfLiteTensor* cell_to_input_weights = in CheckLstmTensorDimensionsAndTypes() local
283 if (cell_to_input_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
284 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
285 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
286 TF_LITE_ENSURE_TYPES_EQ(context, cell_to_input_weights->type, in CheckLstmTensorDimensionsAndTypes()
311 ((cell_to_input_weights != nullptr || use_cifg) && in CheckLstmTensorDimensionsAndTypes()
314 ((cell_to_input_weights == nullptr) && in CheckLstmTensorDimensionsAndTypes()
Dlstm_test.cc1802 const std::vector<float> cell_to_input_weights = {0.3, -0.1, 0.1, -0.2}; in TEST() local
1887 lstm.SetCellToInputWeights(cell_to_input_weights); in TEST()
Dunidirectional_sequence_lstm_test.cc3234 const std::vector<float> cell_to_input_weights = {0.3, -0.1, 0.1, -0.2}; in TEST() local
3319 lstm.SetCellToInputWeights(cell_to_input_weights); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc340 const TfLiteTensor* cell_to_input_weights, in EvalFloat() argument
430 GetTensorData<float>(cell_to_input_weights), in EvalFloat()
492 GetTensorData<float>(cell_to_input_weights), in EvalFloat()
557 const TfLiteTensor* cell_to_input_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.cc272 const TfLiteTensor* cell_to_input_weights, in EvalCalibration() argument
361 GetTensorData<float>(cell_to_input_weights), in EvalCalibration()
423 GetTensorData<float>(cell_to_input_weights), in EvalCalibration()
513 const TfLiteTensor* cell_to_input_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.cc48 std::initializer_list<float> cell_to_input_weights, in LSTMOpModel() argument
92 AddConstInput({weight_type, {n_cell}}, cell_to_input_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc155 {lstm.input_cell_state(), lstm.cell_to_input_weights()}), in LoadForLSTMOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3710 // Ignore input 9 (cell_to_input_weights) for LSTM with CIFG.
3715 TypeIsPred<"cell_to_input_weights", NoneType>]>]>]>>;
3732 TypeIsPred<"cell_to_input_weights", NoneType>,
3801 TFL_OperandIsNoneOrHasRank<9, 1>, // cell_to_input_weights
3847 TFL_TensorOfOrNone<[F32, QI8, QI16]>:$cell_to_input_weights,
3932 TFL_OperandIsNoneOrHasRank<9, 1>, // cell_to_input_weights
3971 TFL_TensorOfOrNone<[F32, QI8]>:$cell_to_input_weights,