Home
last modified time | relevance | path

Searched refs:NodeContext (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc562 struct NodeContext { struct in tensorflow::grappler::SymbolicShapeRefiner
573 NodeContext* GetNodeContext(const NodeDef* node) { in GetNodeContext()
746 NodeContext* ctx = GetNodeContext(node); in UpdateNode()
766 NodeContext* src_ctx = GetNodeContext(src); in UpdateNode()
1071 NodeContext& node_ctx = node_to_context_[node]; in AddNode()
1131 bool AllOutputValuesKnown(NodeContext* c) { in AllOutputValuesKnown()
1158 bool AllInputValuesKnown(NodeContext* c) { in AllInputValuesKnown()
1179 bool ShouldUpdateOutputShapesAndValues(NodeContext* c, int64 max_size) { in ShouldUpdateOutputShapesAndValues()
1229 void CreateInputTensors(NodeContext* c, in CreateInputTensors()
1272 Status UpdateOutputShapesAndValues(const NodeDef& node, NodeContext* c) { in UpdateOutputShapesAndValues()
[all …]