Home
last modified time | relevance | path

Searched refs:GetOptionalInputTensor (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_lstm.cc208 GetOptionalInputTensor(context, node, input_to_input_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
238 GetOptionalInputTensor(context, node, recurrent_to_input_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
282 GetOptionalInputTensor(context, node, cell_to_input_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
291 GetOptionalInputTensor(context, node, cell_to_forget_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
300 GetOptionalInputTensor(context, node, cell_to_output_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
321 GetOptionalInputTensor(context, node, input_gate_bias_tensor); in CheckLstmTensorDimensionsAndTypes()
354 GetOptionalInputTensor(context, node, projection_weights_tensor); in CheckLstmTensorDimensionsAndTypes()
364 GetOptionalInputTensor(context, node, projection_bias_tensor); in CheckLstmTensorDimensionsAndTypes()
490 GetOptionalInputTensor(context, node, kAuxInputTensor); in Prepare()
492 GetOptionalInputTensor(context, node, kFwAuxInputToInputWeightsTensor); in Prepare()
[all …]
H A Dlstm.cc183 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
198 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
213 GetOptionalInputTensor(context, node, kCellToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
215 GetOptionalInputTensor(context, node, kCellToForgetWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
217 GetOptionalInputTensor(context, node, kCellToOutputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
220 GetOptionalInputTensor(context, node, kInputLayerNormCoefficientsTensor); in PopulateQuantizedLstmParams8x8_16()
222 GetOptionalInputTensor(context, node, kForgetLayerNormCoefficientsTensor); in PopulateQuantizedLstmParams8x8_16()
224 GetOptionalInputTensor(context, node, kCellLayerNormCoefficientsTensor); in PopulateQuantizedLstmParams8x8_16()
226 GetOptionalInputTensor(context, node, kOutputLayerNormCoefficientsTensor); in PopulateQuantizedLstmParams8x8_16()
229 GetOptionalInputTensor(context, node, kProjectionWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
[all …]
H A Dunidirectional_sequence_lstm.cc96 const TfLiteTensor* input_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
113 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
131 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
133 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
135 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
138 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
140 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
142 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
144 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
147 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
[all …]
H A Dbidirectional_sequence_rnn.cc130 GetOptionalInputTensor(context, node, kAuxInputTensor); in Prepare()
132 GetOptionalInputTensor(context, node, kFwAuxWeightsTensor); in Prepare()
134 GetOptionalInputTensor(context, node, kBwAuxWeightsTensor); in Prepare()
741 GetOptionalInputTensor(context, node, kAuxInputTensor); in Eval()
743 GetOptionalInputTensor(context, node, kFwAuxWeightsTensor); in Eval()
745 GetOptionalInputTensor(context, node, kBwAuxWeightsTensor); in Eval()
H A Dsvdf.cc110 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
304 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Eval()
H A Dkernel_util.h97 const TfLiteTensor* GetOptionalInputTensor(const TfLiteContext* context,
H A Dtranspose_conv.cc275 bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
608 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
H A Dpad.cc46 constant_values = GetOptionalInputTensor(context, node, 2); in PadContext()
H A Dkernel_util.cc147 const TfLiteTensor* GetOptionalInputTensor(const TfLiteContext* context, in GetOptionalInputTensor() function
H A Dfully_connected.cc212 ? GetOptionalInputTensor(context, node, kBiasTensor) in PrepareImpl()
1258 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc551 const TfLiteTensor* input_to_input_weights = GetOptionalInputTensor( in lstm_eval()
560 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in lstm_eval()
569 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in lstm_eval()
571 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval()
573 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in lstm_eval()
576 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
579 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
582 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
585 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
589 const TfLiteTensor* input_gate_bias = GetOptionalInputTensor( in lstm_eval()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc473 const TfLiteTensor* input_to_input_weights = GetOptionalInputTensor( in lstm_eval()
493 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in lstm_eval()
514 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in lstm_eval()
516 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval()
518 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in lstm_eval()
521 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
524 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
527 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
530 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
534 const TfLiteTensor* input_gate_bias = GetOptionalInputTensor( in lstm_eval()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder_helper.cc205 GetOptionalInputTensor(context, tflite_node, i); in GetNumberOfRuntimeInputsForNode()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/kernels/libtflite_kernel_utils/android_arm64_armv8-2a_cortex-a55_static_apex30_sabi/obj/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.sdump5381 "function_name" : "tflite::GetOptionalInputTensor",