Home
last modified time | relevance | path

Searched refs:GetNodeAndRegistration (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc307 context->GetNodeAndRegistration = [](struct TfLiteContext*, int node_index, in TEST()
342 context->GetNodeAndRegistration(context, ops_to_replace->data[0], &node, in TEST()
366 context->GetNodeAndRegistration = [](struct TfLiteContext*, int node_index, in TEST()
400 context->GetNodeAndRegistration(context, ops_to_replace->data[0], &node, in TEST()
429 context->GetNodeAndRegistration = [](struct TfLiteContext*, int node_index, in TEST()
454 context->GetNodeAndRegistration(context, kGreaterOpIndex, &node, in TEST()
543 context->GetNodeAndRegistration = [](struct TfLiteContext*, int node_index, in TEST()
710 context->GetNodeAndRegistration = [](struct TfLiteContext*, int node_index, in TEST()
930 context->GetNodeAndRegistration = [](struct TfLiteContext*, int node_index, in TEST()
969 context->GetNodeAndRegistration(context, ops_to_replace->data[0], &node, in TEST()
[all …]
Dmodel_builder_helper.h40 absl::Status GetNodeAndRegistration(TfLiteContext* context, int node_id,
Dmodel_builder_helper.cc68 absl::Status GetNodeAndRegistration(TfLiteContext* context, int node_id, in GetNodeAndRegistration() function
71 if (context->GetNodeAndRegistration(context, node_id, tflite_node, in GetNodeAndRegistration()
Dmodel_builder.cc2604 RETURN_IF_ERROR(GetNodeAndRegistration( in BuildModel()
2635 RETURN_IF_ERROR(GetNodeAndRegistration( in BuildModel()
/external/tensorflow/tensorflow/lite/delegates/
Dutils.cc130 status = context_->GetNodeAndRegistration(context_, node_id, &node, in PrepareSupportedNodes()
174 const auto status = context_->GetNodeAndRegistration(context_, node_id, in GetNodesOfFirstNLargestPartitionsImpl()
272 TfLiteStatus status = context_->GetNodeAndRegistration( in RemapFp16InputTensors()
Ddelegate_test_util.cc152 context->GetNodeAndRegistration(context, node_index, &node, &reg); in SimpleDelegate()
167 context->GetNodeAndRegistration(context, node_index, &node, &reg); in SimpleDelegate()
Dutils_test.cc113 this->GetNodeAndRegistration = MockGetNodeAndRegistration;
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc1174 TfLiteStatus Subgraph::GetNodeAndRegistration( in GetNodeAndRegistration() function in tflite::Subgraph
1186 TfLiteStatus Subgraph::GetNodeAndRegistration( in GetNodeAndRegistration() function in tflite::Subgraph
1190 ->GetNodeAndRegistration(node_index, node, registration); in GetNodeAndRegistration()
1354 context_.GetNodeAndRegistration = GetNodeAndRegistration; in SwitchToDelegateContext()
1362 context_.GetNodeAndRegistration = [](struct TfLiteContext* context, in SwitchToKernelContext()
Dsubgraph.h506 TfLiteStatus GetNodeAndRegistration(int node_index, TfLiteNode** node,
511 static TfLiteStatus GetNodeAndRegistration(struct TfLiteContext*,
/external/tensorflow/tensorflow/lite/c/
Dcommon.h664 TfLiteStatus (*GetNodeAndRegistration)( member
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc257 context->GetNodeAndRegistration(context, node_index, &node, &reg)); in BuildGraph()
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_delegate_kernel.mm125 TF_LITE_ENSURE_STATUS(context->GetNodeAndRegistration(context, node_index, &node, &reg));
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_version.md259 TF_LITE_ENSURE_STATUS(context->GetNodeAndRegistration(context, node_index, &node, &registration));
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc404 context->GetNodeAndRegistration(context, node_index, &node, &reg); in Init()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc149 if (context->GetNodeAndRegistration(context, node_index, &node, in Create()
265 if (context->GetNodeAndRegistration(context, node_index, &node, in Create()
2954 if (context->GetNodeAndRegistration(context, node_index, &node, in PrepareOpsToDelegate()
3065 if (context->GetNodeAndRegistration(context, producer_index, &node, in PrepareOpsToDelegate()
/external/tensorflow/tensorflow/lite/g3doc/performance/
Dimplementing_delegate.md154 context->GetNodeAndRegistration(context, node_index, &delegated_node,
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc983 context->GetNodeAndRegistration(context, 0, &node, &registration), in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc4208 context->GetNodeAndRegistration(context, node_index, &node, &reg)); in AddOpsAndTensors()
5112 TF_LITE_ENSURE_STATUS(context->GetNodeAndRegistration( in DoPrepare()
/external/tensorflow/
DRELEASE.md78 and `context()`, and then calling the `GetNodeAndRegistration`