Searched refs:TestVirtualScheduler (Results 1 – 1 of 1) sorted by relevance
645 class TestVirtualScheduler : public VirtualScheduler { class647 TestVirtualScheduler(const bool use_static_shapes, in TestVirtualScheduler() function in tensorflow::grappler::__anon653cd6ac0111::TestVirtualScheduler678 scheduler_ = absl::make_unique<TestVirtualScheduler>( in VirtualSchedulerTest()2383 std::unique_ptr<TestVirtualScheduler> scheduler_;3075 scheduler_ = absl::make_unique<TestVirtualScheduler>( in TEST_F()