diff options
Diffstat (limited to 'tensorflow/core/grappler/costs/graph_properties.h')
-rw-r--r-- | tensorflow/core/grappler/costs/graph_properties.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/graph_properties.h b/tensorflow/core/grappler/costs/graph_properties.h index ee279b7e0a..239b5ac244 100644 --- a/tensorflow/core/grappler/costs/graph_properties.h +++ b/tensorflow/core/grappler/costs/graph_properties.h @@ -58,6 +58,12 @@ class GraphProperties { const std::vector<OpInfo::TensorProperties>& GetOutputProperties( const string& node_name) const; + static void FillTensorPropertiesFromContext( + const shape_inference::ShapeHandle&, const DataType&, + shape_inference::InferenceContext*, + std::unordered_map<const shape_inference::Dimension*, int>* dim_ids, + OpInfo::TensorProperties*); + private: // Inputs GrapplerItem item_; |