diff options
author | 2017-10-03 01:04:18 -0700 | |
---|---|---|
committer | 2017-10-03 01:08:11 -0700 | |
commit | 0ea4331690c9f00abfbb634a91520042b7b84a20 (patch) | |
tree | f63f834d30c059055d0b318123e162eb52445688 /tensorflow/tools/graph_transforms/transform_utils.h | |
parent | 263d025fb6dee974eefb30a51372188fb856d6cc (diff) |
Use shape information in constant propagation.
PiperOrigin-RevId: 170818644
Diffstat (limited to 'tensorflow/tools/graph_transforms/transform_utils.h')
-rw-r--r-- | tensorflow/tools/graph_transforms/transform_utils.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/tools/graph_transforms/transform_utils.h b/tensorflow/tools/graph_transforms/transform_utils.h index c0fb492412..47c8aaed2c 100644 --- a/tensorflow/tools/graph_transforms/transform_utils.h +++ b/tensorflow/tools/graph_transforms/transform_utils.h @@ -133,6 +133,9 @@ Status IsGraphValid(const GraphDef& graph_def); Status GetInOutTypes(const NodeDef& node_def, DataTypeVector* inputs, DataTypeVector* outputs); +// Takes a comma-separated string of numbers and parses them into a shape. +Status TensorShapeFromString(const string& shape_string, TensorShape* result); + // This is used to spot particular subgraphs in a larger model. To use it, // create a pattern like: // OpTypePattern pattern({"Conv2D", {{"ResizeBilinear", {{"MirrorPad"}}}}}); |