Searched refs:external_context (Results 1 – 5 of 5) sorted by relevance
30 external_context = nullptr; in TestTfLiteContext()37 TfLiteExternalContext* external_context) { in SetExternalContextImpl()38 static_cast<TestTfLiteContext*>(context)->external_context = in SetExternalContextImpl()39 external_context; in SetExternalContextImpl()44 return static_cast<TestTfLiteContext*>(context)->external_context; in GetExternalContextImpl()47 TfLiteExternalContext* external_context; member53 ASSERT_NE(context.external_context, nullptr); in TEST()54 EXPECT_EQ(context.external_context->type, kTfLiteEigenContext); in TEST()114 ASSERT_NE(context.external_context, nullptr); in TEST()115 context.external_context->Refresh(&context); in TEST()[all …]
102 auto* external_context = static_cast<ExternalCpuBackendContext*>( in GetFromContext() local105 if (external_context == nullptr) { in GetFromContext()112 external_context->internal_backend_context()); in GetFromContext()118 external_context->set_internal_backend_context( in GetFromContext()
23 auto* const external_context = static_cast<ExternalCpuBackendContext*>( in RefreshExternalCpuBackendContext() local25 if (external_context && external_context->internal_backend_context() && in RefreshExternalCpuBackendContext()27 external_context->internal_backend_context()->SetMaxNumThreads( in RefreshExternalCpuBackendContext()
124 ExternalCpuBackendContext* external_context = in ~Interpreter() local128 external_context->internal_backend_context(); in ~Interpreter()
1119 TestExternalContext external_context; in TEST_F() local1120 external_context.Refresh = [](TfLiteContext* context) { in TEST_F()1131 TestExternalContext::Set(context, &external_context); in TEST_F()1132 EXPECT_EQ(TestExternalContext::Get(context), &external_context); in TEST_F()1135 EXPECT_EQ(external_context.num_refreshes, 2); in TEST_F()1138 external_context.num_refreshes = 0; in TEST_F()1141 EXPECT_EQ(external_context.num_refreshes, 0); in TEST_F()1144 EXPECT_EQ(external_context.num_refreshes, 1); in TEST_F()