Searched refs:GetEagerContext (Results 1 – 4 of 4) sorted by relevance
42 tensorflow::EagerContext* GetEagerContext() { return eager_context_; } in GetEagerContext() function
37 EXPECT_NE(data.GetEagerContext(), nullptr); in TEST()
382 op_data_->eager_context = flex_delegate_data->GetEagerContext(); in Init()
50 Status GetEagerContext(const uint64 context_id, EagerContext** ctx) { in GetEagerContext() function in tensorflow::eager::__anon58c807940111::TestEagerServiceImpl751 TF_ASSERT_OK(eager_service_impl_.GetEagerContext(context_id_, &ctx)); in Init()848 TF_ASSERT_OK(eager_service_impl_.GetEagerContext(context_id_, &ctx)); in TEST_F()909 TF_ASSERT_OK(eager_service_impl_.GetEagerContext(context_id_, &ctx)); in TEST_F()949 TF_ASSERT_OK(eager_service_impl_.GetEagerContext(context_id_, &ctx)); in TEST_F()997 TF_ASSERT_OK(eager_service_impl_.GetEagerContext(context_id_, &ctx)); in TEST_F()