experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Misc. cleanup in grappler.

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/grappler/clusters/virtual_cluster.h
View File

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

4
tensorflow/core/grappler/costs/robust_stats.h
View File

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

3
tensorflow/core/grappler/costs/virtual_placer.h
View File

@ -30,7 +30,8 @@ class Cluster;
// The virtual placer emulates the behavior of the TF placer. // The virtual placer emulates the behavior of the TF placer.
class VirtualPlacer { class VirtualPlacer {
public: 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; const DeviceProperties& get_device(const NodeDef& node) const;

3
tensorflow/core/grappler/optimizers/scoped_allocator_optimizer.h
View File

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