Searched refs:tflite_node (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 80 absl::Status RetrieveBuiltinData(const TfLiteNode* tflite_node, in RetrieveBuiltinData() argument
82 *tf_options = static_cast<const ParamsT*>(tflite_node->builtin_data); in RetrieveBuiltinData()
90 absl::Status RetrieveCustomInitialData(const TfLiteNode* tflite_node, in RetrieveCustomInitialData() argument
92 *tf_options = static_cast<const ParamsT*>(tflite_node->custom_initial_data); in RetrieveCustomInitialData()
179 OperationType operation_type, const TfLiteNode* tflite_node, in MaybeFuseActivationForElementwiseNode() argument
185 if (RetrieveBuiltinData(tflite_node, &tf_options).ok()) { in MaybeFuseActivationForElementwiseNode()
192 if (RetrieveBuiltinData(tflite_node, &tf_options).ok()) { in MaybeFuseActivationForElementwiseNode()
199 if (RetrieveBuiltinData(tflite_node, &tf_options).ok()) { in MaybeFuseActivationForElementwiseNode()
206 if (RetrieveBuiltinData(tflite_node, &tf_options).ok()) { in MaybeFuseActivationForElementwiseNode()
279 const TfLiteNode* tflite_node, in IsSupported() argument
[all …]
|
| D | model_builder_helper.cc | 69 TfLiteNode** tflite_node, in GetNodeAndRegistration() argument
71 if (context->GetNodeAndRegistration(context, node_id, tflite_node, in GetNodeAndRegistration()
201 const TfLiteNode* tflite_node) { in GetNumberOfRuntimeInputsForNode() argument
203 for (int i = 0; i < NumInputs(tflite_node); i++) { in GetNumberOfRuntimeInputsForNode()
205 GetOptionalInputTensor(context, tflite_node, i); in GetNumberOfRuntimeInputsForNode()
214 const TfLiteNode* tflite_node) { in GetNumberOfConstInputsForNode() argument
215 return NumInputs(tflite_node) - in GetNumberOfConstInputsForNode()
216 GetNumberOfRuntimeInputsForNode(context, tflite_node); in GetNumberOfConstInputsForNode()
220 const TfLiteNode* tflite_node, in CheckInputsOutputs() argument
223 GetNumberOfRuntimeInputsForNode(context, tflite_node); in CheckInputsOutputs()
[all …]
|
| D | model_builder_helper.h | 41 TfLiteNode** tflite_node,
61 const TfLiteNode* tflite_node);
64 const TfLiteNode* tflite_node);
67 const TfLiteNode* tflite_node,
71 const TfLiteNode* tflite_node,
|
| D | unimplemented_operation_parser.h | 36 const TfLiteNode* tflite_node, in IsSupported() argument
41 absl::Status Parse(const TfLiteNode* tflite_node, in Parse() argument
|
| D | operation_parser.h | 36 virtual absl::Status Parse(const TfLiteNode* tflite_node,
42 const TfLiteNode* tflite_node,
|
| D | lstm_parser.h | 28 const TfLiteNode* tflite_node, const TfLiteRegistration* registration,
|
| D | lstm_parser.cc | 462 const TfLiteNode* tflite_node, const TfLiteRegistration* registration, in ParseLSTMAttributes() argument
465 const bool has_cifg = HasCifg(tflite_node); in ParseLSTMAttributes()
466 const bool has_peephole = HasPeephole(tflite_node); in ParseLSTMAttributes()
467 const bool has_normalization = HasNormalization(tflite_node); in ParseLSTMAttributes()
468 const bool has_projection = HasProjection(tflite_node); in ParseLSTMAttributes()
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_signature.cc | 92 const TfLiteNode* tflite_node) { in GetOpSignatureTensorSpecs() argument
|