HomeSort by relevance Sort by last modified time
    Searched refs:GetOptionalInputTensor (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/kernels/
lstm.cc 86 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
106 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor);
140 GetOptionalInputTensor(context, node, kCellToInputWeightsTensor);
147 GetOptionalInputTensor(context, node, kCellToForgetWeightsTensor);
154 GetOptionalInputTensor(context, node, kCellToOutputWeightsTensor);
173 GetOptionalInputTensor(context, node, kInputGateBiasTensor);
196 GetOptionalInputTensor(context, node, kProjectionWeightsTensor);
204 GetOptionalInputTensor(context, node, kProjectionBiasTensor);
285 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
311 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor)
    [all...]
unidirectional_sequence_lstm.cc 86 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
106 GetOptionalInputTensor(context, node, kRecurrentToInputWeightsTensor);
140 GetOptionalInputTensor(context, node, kCellToInputWeightsTensor);
147 GetOptionalInputTensor(context, node, kCellToForgetWeightsTensor);
154 GetOptionalInputTensor(context, node, kCellToOutputWeightsTensor);
173 GetOptionalInputTensor(context, node, kInputGateBiasTensor);
196 GetOptionalInputTensor(context, node, kProjectionWeightsTensor);
204 GetOptionalInputTensor(context, node, kProjectionBiasTensor);
287 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor);
313 GetOptionalInputTensor(context, node, kInputToInputWeightsTensor)
    [all...]
kernel_util.h 46 inline TfLiteTensor* GetOptionalInputTensor(TfLiteContext* context,
svdf.cc 77 TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor);
137 TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor);
fully_connected.cc 89 TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor);
245 TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor);

Completed in 239 milliseconds