Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
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 …]
Dunidirectional_sequence_lstm.cc95 const TfLiteTensor* input_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
112 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
130 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
132 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
134 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
137 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
139 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
141 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
143 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
146 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
[all …]
Dlstm.cc182 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
197 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
212 GetOptionalInputTensor(context, node, kCellToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
214 GetOptionalInputTensor(context, node, kCellToForgetWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
216 GetOptionalInputTensor(context, node, kCellToOutputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
219 GetOptionalInputTensor(context, node, kInputLayerNormCoefficientsTensor); in PopulateQuantizedLstmParams8x8_16()
221 GetOptionalInputTensor(context, node, kForgetLayerNormCoefficientsTensor); in PopulateQuantizedLstmParams8x8_16()
223 GetOptionalInputTensor(context, node, kCellLayerNormCoefficientsTensor); in PopulateQuantizedLstmParams8x8_16()
225 GetOptionalInputTensor(context, node, kOutputLayerNormCoefficientsTensor); in PopulateQuantizedLstmParams8x8_16()
228 GetOptionalInputTensor(context, node, kProjectionWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
[all …]
Dbidirectional_sequence_rnn.cc130 GetOptionalInputTensor(context, node, kAuxInputTensor); in Prepare()
132 GetOptionalInputTensor(context, node, kFwAuxWeightsTensor); in Prepare()
134 GetOptionalInputTensor(context, node, kBwAuxWeightsTensor); in Prepare()
739 GetOptionalInputTensor(context, node, kAuxInputTensor); in Eval()
741 GetOptionalInputTensor(context, node, kFwAuxWeightsTensor); in Eval()
743 GetOptionalInputTensor(context, node, kBwAuxWeightsTensor); in Eval()
Dsvdf.cc110 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
302 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Eval()
Dkernel_util.h94 const TfLiteTensor* GetOptionalInputTensor(const TfLiteContext* context,
Dpad.cc46 constant_values = GetOptionalInputTensor(context, node, 2); in PadContext()
Dtranspose_conv.cc275 bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
578 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
Dfully_connected.cc208 ? GetOptionalInputTensor(context, node, kBiasTensor) in PrepareImpl()
975 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
Dkernel_util.cc146 const TfLiteTensor* GetOptionalInputTensor(const TfLiteContext* context, in GetOptionalInputTensor() function
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
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 …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
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 …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_helper.cc203 GetOptionalInputTensor(context, tflite_node, i); in GetNumberOfRuntimeInputsForNode()