Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc618 const absl::flat_hash_map<string, absl::flat_hash_set<int>>& fed_ports, in SymbolicShapeRefiner() argument
622 fed_ports_(fed_ports), in SymbolicShapeRefiner()
2477 absl::flat_hash_map<string, absl::flat_hash_set<int>> fed_ports; in InferStatically() local
2481 fed_ports[tensor_id.node()].insert(tensor_id.index()); in InferStatically()
2527 if (fed_ports.find(node.name()) != fed_ports.end()) { in InferStatically()
2562 graph_view, fed_ports, aggressive_shape_inference); in InferStatically()
2587 if (fed_ports.find(node.name()) != fed_ports.end()) { in InferStatically()