Misc. cleanup in grappler.

PiperOrigin-RevId: 256264467
This commit is contained in:
A. Unique TensorFlower 2019-07-02 16:38:31 -07:00 committed by TensorFlower Gardener
parent 237ce08c89
commit 576d32c57d
4 changed files with 9 additions and 6 deletions

View File

@ -33,11 +33,12 @@ namespace grappler {
// actual graphs.
class VirtualCluster : public Cluster {
public:
VirtualCluster(const std::unordered_map<string, DeviceProperties>& devices);
explicit VirtualCluster(
const std::unordered_map<string, DeviceProperties>& devices);
VirtualCluster(const std::unordered_map<string, DeviceProperties>& devices,
std::unique_ptr<OpLevelCostEstimator> node_estimator,
std::unique_ptr<ReadyNodeManager> node_manager);
VirtualCluster(const DeviceSet* device_set);
explicit VirtualCluster(const DeviceSet* device_set);
~VirtualCluster() override;

View File

@ -21,8 +21,8 @@ namespace tensorflow {
namespace grappler {
class RobustStats {
public:
RobustStats(const std::vector<double>& values);
RobustStats(std::vector<double>&& values);
explicit RobustStats(const std::vector<double>& values);
explicit RobustStats(std::vector<double>&& values);
double lo() const { return lo_; }
double hi() const { return hi_; }

View File

@ -30,7 +30,8 @@ class Cluster;
// The virtual placer emulates the behavior of the TF placer.
class VirtualPlacer {
public:
VirtualPlacer(const std::unordered_map<string, DeviceProperties>& devices);
explicit VirtualPlacer(
const std::unordered_map<string, DeviceProperties>& devices);
const DeviceProperties& get_device(const NodeDef& node) const;

View File

@ -25,8 +25,9 @@ limitations under the License.
#include "tensorflow/core/protobuf/rewriter_config.pb.h"
namespace tensorflow {
namespace grappler {
class Graph;
namespace grappler {
class GraphProperties;
class NodeMap;
class ScopedAllocatorOptimizer;