Searched refs:SetExternalContext (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | eigen_support.cc | 166 context->SetExternalContext(context, kTfLiteEigenContext, ptr); in IncrementUsageCounter()
180 context->SetExternalContext(context, kTfLiteEigenContext, nullptr); in DecrementUsageCounter()
|
| D | eigen_support_test.cc | 32 SetExternalContext = SetExternalContextImpl; in TestTfLiteContext()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.cc | 138 void Interpreter::SetExternalContext(TfLiteExternalContextType type, in SetExternalContext() function in tflite::Interpreter
160 primary_subgraph().SetExternalContext(type, ctx); in SetExternalContext()
|
| D | interpreter.h | 598 void SetExternalContext(TfLiteExternalContextType type,
680 static void SetExternalContext(struct TfLiteContext* context,
|
| D | interpreter_test.cc | 1110 context->SetExternalContext(context, kType, value); in Set()
1168 interpreter.SetExternalContext(kTfLiteCpuBackendContext, in TEST_F()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 254 void SetExternalContext(TfLiteExternalContextType type,
554 static void SetExternalContext(struct TfLiteContext* context,
|
| D | subgraph.cc | 229 context_.SetExternalContext = SetExternalContext; in Subgraph()
453 void Subgraph::SetExternalContext(TfLiteExternalContextType type, in SetExternalContext() function in tflite::Subgraph
460 void Subgraph::SetExternalContext(struct TfLiteContext* context, in SetExternalContext() function in tflite::Subgraph
463 return static_cast<Subgraph*>(context->impl_)->SetExternalContext(type, ctx); in SetExternalContext()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 685 void (*SetExternalContext)(struct TfLiteContext*, TfLiteExternalContextType, member
|
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| D | benchmark_tflite_model.cc | 605 interpreter_->SetExternalContext(kTfLiteCpuBackendContext, in InitInterpreter()
|