Searched refs:TestVirtualScheduler (Results 1 – 1 of 1) sorted by relevance
31 class TestVirtualScheduler : public VirtualScheduler { class33 TestVirtualScheduler(const bool use_static_shapes, in TestVirtualScheduler() function in tensorflow::grappler::TestVirtualScheduler73 scheduler_ = absl::make_unique<TestVirtualScheduler>( in VirtualSchedulerTest()1709 std::unique_ptr<TestVirtualScheduler> scheduler_;