Searched refs:virtual_placer_ (Results 1 – 3 of 3) sorted by relevance
53 std::unique_ptr<VirtualPlacer> virtual_placer_;
410 virtual_placer_(virtual_placer), in GraphProcessor()434 device_name = virtual_placer_.get_canonical_device_name(*node); in AddNodePermConst()458 device_name = virtual_placer_.get_canonical_device_name(*node); in AddNodeConstScalar()471 const VirtualPlacer& virtual_placer_; member in tensorflow::grappler::__anon49c8f8520111::GraphProcessor586 device_name = virtual_placer_.get_canonical_device_name(*node_); in IsOnGPU()2068 virtual_placer_, nodes_to_preserve_, in Expand()2116 virtual_placer_, nodes_to_preserve_, in Expand()2246 DataLayoutOptimizer layout_optimizer(graph_properties, *virtual_placer_, in Tune()2271 virtual_placer_.reset(new VirtualPlacer(cluster->GetDevices())); in Optimize()
932 : virtual_placer_(cluster->GetDevices()), in AutoMixedPrecisionImpl()998 VirtualPlacer virtual_placer_; member in tensorflow::grappler::__anon69ea7d4c0111::AutoMixedPrecisionImpl1043 string device_name = virtual_placer_.get_canonical_device_name(node); in NodeHasF16KernelForTypeAttr()1122 device_name = virtual_placer_.get_canonical_device_name(node); in IsOnDevice()1163 return GetDeviceGPUArch(virtual_placer_.get_device(node)) >= kMinGPUArch; in IsOnSuitableGPUArch()