| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm.cc | 208 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 …]
|
| D | unidirectional_sequence_lstm.cc | 91 const TfLiteTensor* input_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
108 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
126 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
128 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
130 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
133 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
135 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
137 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
139 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
142 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16()
[all …]
|
| D | lstm.cc | 182 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 …]
|
| D | bidirectional_sequence_rnn.cc | 130 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()
|
| D | kernel_util.h | 94 const TfLiteTensor* GetOptionalInputTensor(const TfLiteContext* context,
|
| D | svdf.cc | 109 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
294 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Eval()
|
| D | pad.cc | 46 constant_values = GetOptionalInputTensor(context, node, 2); in PadContext()
|
| D | transpose_conv.cc | 275 bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
578 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
|
| D | kernel_util.cc | 139 const TfLiteTensor* GetOptionalInputTensor(const TfLiteContext* context, in GetOptionalInputTensor() function
|
| D | fully_connected.cc | 208 ? GetOptionalInputTensor(context, node, kBiasTensor) in PrepareImpl()
921 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 539 const TfLiteTensor* input_to_input_weights = GetOptionalInputTensor( in lstm_eval()
548 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in lstm_eval()
557 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in lstm_eval()
559 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval()
561 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in lstm_eval()
564 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
567 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
570 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
573 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
577 const TfLiteTensor* input_gate_bias = GetOptionalInputTensor( in lstm_eval()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 472 const TfLiteTensor* input_to_input_weights = GetOptionalInputTensor( in lstm_eval()
492 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in lstm_eval()
513 const TfLiteTensor* cell_to_input_weights = GetOptionalInputTensor( in lstm_eval()
515 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval()
517 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in lstm_eval()
520 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
523 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
526 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
529 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval()
533 const TfLiteTensor* input_gate_bias = GetOptionalInputTensor( in lstm_eval()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | fully_connected.cc | 52 GetOptionalInputTensor(context, node, kFullyConnectedBiasTensor); in Prepare()
|
| D | conv_common.cc | 101 GetOptionalInputTensor(context, node, kConvBiasTensor); in CalculateOpDataConv()
|
| D | transpose_conv.cc | 96 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
|
| D | depthwise_conv.cc | 86 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
|
| D | svdf.cc | 215 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | fully_connected.cc | 163 GetOptionalInputTensor(context, node, kFullyConnectedBiasTensor); in Prepare()
|
| D | svdf.cc | 338 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/ |
| D | depthwise_conv.cc | 121 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
154 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
|
| D | conv.cc | 131 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
163 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
|
| D | fully_connected.cc | 118 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | fully_connected.cc | 62 GetOptionalInputTensor(context, node, kFullyConnectedBiasTensor); in Prepare()
|
| D | svdf.cc | 303 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_helper.cc | 203 GetOptionalInputTensor(context, tflite_node, i); in GetNumberOfRuntimeInputsForNode()
|