Searched refs:VirtualPlacer (Results 1 – 15 of 15) sorted by relevance
36 VirtualPlacer placer(devices); in TEST()66 VirtualPlacer placer(devices); in TEST()96 VirtualPlacer placer(devices); in TEST()126 VirtualPlacer placer(devices); in TEST()148 VirtualPlacer placer(devices); in TEST()160 VirtualPlacer placer(devices); in GetDefaultDeviceName()207 std::unique_ptr<VirtualPlacer> placer(new VirtualPlacer(devices)); in TEST()238 placer.reset(new VirtualPlacer(cluster->GetDevices())); in TEST()258 VirtualPlacer placer(devices); in TEST()274 VirtualPlacer placer(devices); in TEST()[all …]
26 VirtualPlacer::VirtualPlacer( in VirtualPlacer() function in tensorflow::grappler::VirtualPlacer115 const DeviceProperties& VirtualPlacer::get_device(const NodeDef& node) const { in get_device()124 string VirtualPlacer::get_canonical_device_name(const NodeDef& node) const { in get_canonical_device_name()142 string VirtualPlacer::to_lfqn_or_empty(const string& device_name) const { in to_lfqn_or_empty()
31 class VirtualPlacer {33 explicit VirtualPlacer(
325 std::unique_ptr<VirtualPlacer> placer);445 std::unique_ptr<VirtualPlacer> placer_;456 std::unique_ptr<VirtualPlacer> placer);
53 std::unique_ptr<VirtualPlacer> placer,
129 absl::make_unique<VirtualPlacer>(cluster->GetDevices())); in AnalyticalCostEstimator()135 std::unique_ptr<VirtualPlacer> placer, bool use_static_shapes, in AnalyticalCostEstimator()
387 std::unique_ptr<VirtualPlacer> placer) in SchedulerState()1319 std::unique_ptr<VirtualPlacer> placer) in VirtualScheduler()
653 absl::make_unique<VirtualPlacer>(cluster->GetDevices())) { in TestVirtualScheduler()
34 const VirtualPlacer& placer, const NodeDef& node) { in PredictExecutionTime()102 VirtualPlacer placer(cluster->GetDevices()); in EstimateEarliestExecutionTimes()173 VirtualPlacer placer(cluster->GetDevices()); in EstimateRequiredTimes()
53 std::unique_ptr<VirtualPlacer> virtual_placer_;
72 std::unique_ptr<const VirtualPlacer> virtual_placer;607 string GetDeviceName(const VirtualPlacer* virtual_placer, const NodeDef& node);
406 const VirtualPlacer& virtual_placer, in GraphProcessor()471 const VirtualPlacer& virtual_placer_;480 const VirtualPlacer& virtual_placer, in OptimizeContext()496 const VirtualPlacer& virtual_placer;2030 const VirtualPlacer& virtual_placer, in DataLayoutOptimizer()2271 virtual_placer_.reset(new VirtualPlacer(cluster->GetDevices())); in Optimize()
244 absl::make_unique<const VirtualPlacer>(cluster->GetDevices()); in InitializeTransposeContext()1954 string GetDeviceName(const VirtualPlacer* virtual_placer, const NodeDef& node) { in GetDeviceName()
998 VirtualPlacer virtual_placer_;
1246 VirtualPlacer virtual_placer(virtual_cluster_->GetDevices()); in TEST_F()