Home
last modified time | relevance | path

Searched refs:tflite_node (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc76 absl::Status RetrieveBuiltinData(const TfLiteNode* tflite_node, in RetrieveBuiltinData() argument
78 *tf_options = static_cast<const ParamsT*>(tflite_node->builtin_data); in RetrieveBuiltinData()
175 const TfLiteNode* tflite_node, in IsSupported() argument
178 if (tflite_node->inputs->size != 2) { in IsSupported()
184 return RetrieveBuiltinData(tflite_node, &tf_options); in IsSupported()
187 absl::Status Parse(const TfLiteNode* tflite_node, in Parse() argument
201 RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &tf_options)); in Parse()
209 const TfLiteNode* tflite_node, in IsSupported() argument
211 return CheckInputsOutputs(context, tflite_node, in IsSupported()
215 absl::Status Parse(const TfLiteNode* tflite_node, in Parse() argument
[all …]
Dmodel_builder_helper.cc69 TfLiteNode** tflite_node, in GetNodeAndRegistration() argument
71 if (context->GetNodeAndRegistration(context, node_id, tflite_node, in GetNodeAndRegistration()
199 const TfLiteNode* tflite_node) { in GetNumberOfRuntimeInputsForNode() argument
201 for (int i = 0; i < NumInputs(tflite_node); i++) { in GetNumberOfRuntimeInputsForNode()
203 GetOptionalInputTensor(context, tflite_node, i); in GetNumberOfRuntimeInputsForNode()
212 const TfLiteNode* tflite_node) { in GetNumberOfConstInputsForNode() argument
213 return NumInputs(tflite_node) - in GetNumberOfConstInputsForNode()
214 GetNumberOfRuntimeInputsForNode(context, tflite_node); in GetNumberOfConstInputsForNode()
218 const TfLiteNode* tflite_node, in CheckInputsOutputs() argument
221 GetNumberOfRuntimeInputsForNode(context, tflite_node); in CheckInputsOutputs()
[all …]
Dmodel_builder_helper.h41 TfLiteNode** tflite_node,
61 const TfLiteNode* tflite_node);
64 const TfLiteNode* tflite_node);
67 const TfLiteNode* tflite_node,
71 const TfLiteNode* tflite_node,
Doperation_parser.h36 virtual absl::Status Parse(const TfLiteNode* tflite_node,
42 const TfLiteNode* tflite_node,
59 const TfLiteNode* tflite_node, int idx);
Doperation_parser.cc65 const TfLiteNode* tflite_node, int idx) { in CheckTensorIsAvailable() argument
67 if (idx >= tflite_node->inputs->size) { in CheckTensorIsAvailable()
70 idx, tflite_node->inputs->size)); in CheckTensorIsAvailable()
Dlstm_parser.h28 const TfLiteNode* tflite_node, const TfLiteRegistration* registration,
Dlstm_parser.cc444 const TfLiteNode* tflite_node, const TfLiteRegistration* registration, in ParseLSTMAttributes() argument
447 const bool has_cifg = HasCifg(tflite_node); in ParseLSTMAttributes()
448 const bool has_peephole = HasPeephole(tflite_node); in ParseLSTMAttributes()
449 const bool has_normalization = HasNormalization(tflite_node); in ParseLSTMAttributes()
450 const bool has_projection = HasProjection(tflite_node); in ParseLSTMAttributes()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/default/
Dcustom_parsers.cc40 const TfLiteNode* tflite_node, in IsSupported() argument
45 absl::Status Parse(const TfLiteNode* tflite_node, in Parse() argument