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 | 141 void Interpreter::SetExternalContext(TfLiteExternalContextType type, in SetExternalContext() function in tflite::Interpreter
163 primary_subgraph().SetExternalContext(type, ctx); in SetExternalContext()
|
| D | interpreter.h | 665 void SetExternalContext(TfLiteExternalContextType type,
743 static void SetExternalContext(struct TfLiteContext* context,
|
| D | interpreter_test.cc | 1180 context->SetExternalContext(context, kType, value); in Set()
1244 interpreter.SetExternalContext(kTfLiteCpuBackendContext, in TEST_F()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 269 void SetExternalContext(TfLiteExternalContextType type,
650 static void SetExternalContext(struct TfLiteContext* context,
|
| D | subgraph.cc | 257 context_.SetExternalContext = SetExternalContext; in Subgraph()
485 void Subgraph::SetExternalContext(TfLiteExternalContextType type, in SetExternalContext() function in tflite::Subgraph
492 void Subgraph::SetExternalContext(struct TfLiteContext* context, in SetExternalContext() function in tflite::Subgraph
495 return static_cast<Subgraph*>(context->impl_)->SetExternalContext(type, ctx); in SetExternalContext()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 748 void (*SetExternalContext)(struct TfLiteContext*, TfLiteExternalContextType, member
|
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| D | benchmark_tflite_model.cc | 684 interpreter_->SetExternalContext(kTfLiteCpuBackendContext, in InitInterpreter()
|