Searched refs:VirtualScheduler (Results 1 – 5 of 5) sorted by relevance
259 VirtualScheduler::VirtualScheduler(const bool use_static_shapes, in VirtualScheduler() function in tensorflow::grappler::VirtualScheduler274 Status VirtualScheduler::Init(const GrapplerItem* item) { in Init()466 void VirtualScheduler::MaybeUpdateInputOutput(const NodeDef* node) { in MaybeUpdateInputOutput()515 float VirtualScheduler::Round2(const float x) const { in Round2()521 bool VirtualScheduler::IsPersistentNode(const NodeDef* node) const { in IsPersistentNode()526 string VirtualScheduler::DeviceName(const NodeDef* node) const { in DeviceName()530 string VirtualScheduler::SanitizedDeviceName(const NodeDef* node) const { in SanitizedDeviceName()537 string VirtualScheduler::ChannelDeviceName(const NodeDef* from, in ChannelDeviceName()544 std::pair<const NodeDef*, const NodeDef*> VirtualScheduler::CreateSendRecv( in CreateSendRecv()618 OpContext VirtualScheduler::GetCurrNode() const { in GetCurrNode()[all …]
63 const VirtualScheduler* GetScheduler() const { return scheduler_.get(); } in GetScheduler()70 std::unique_ptr<VirtualScheduler> scheduler_;
261 class VirtualScheduler {264 VirtualScheduler(const bool use_static_shapes,
125 scheduler_ = absl::make_unique<VirtualScheduler>( in AnalyticalCostEstimator()
31 class TestVirtualScheduler : public VirtualScheduler {36 : VirtualScheduler(use_static_shapes, use_aggressive_shape_inference, in TestVirtualScheduler()