Searched refs:VirtualScheduler (Results 1 – 5 of 5) sorted by relevance
67 const VirtualScheduler* GetScheduler() const { return scheduler_.get(); } in GetScheduler()73 std::unique_ptr<VirtualScheduler> scheduler_;
450 class VirtualScheduler {453 VirtualScheduler(const bool use_static_shapes,463 VirtualScheduler(ReadyNodeManager* ready_nodes,465 virtual ~VirtualScheduler();
1313 VirtualScheduler::~VirtualScheduler() {} in ~VirtualScheduler()1315 VirtualScheduler::VirtualScheduler(const bool use_static_shapes, in VirtualScheduler() function in tensorflow::grappler::VirtualScheduler1325 VirtualScheduler::VirtualScheduler( in VirtualScheduler() function in tensorflow::grappler::VirtualScheduler1330 Status VirtualScheduler::Init(const GrapplerItem* item) { in Init()1347 OpContext VirtualScheduler::GetCurrNode() { in GetCurrNode()1352 bool VirtualScheduler::MarkCurrNodeExecuted(const Costs& node_costs) { in MarkCurrNodeExecuted()
126 scheduler_ = absl::make_unique<VirtualScheduler>( in AnalyticalCostEstimator()141 scheduler_ = absl::make_unique<VirtualScheduler>( in AnalyticalCostEstimator()
645 class TestVirtualScheduler : public VirtualScheduler {650 : VirtualScheduler( in TestVirtualScheduler()