Searched refs:fed_ports (Results 1 – 1 of 1) sorted by relevance
618 const absl::flat_hash_map<string, absl::flat_hash_set<int>>& fed_ports, in SymbolicShapeRefiner() argument622 fed_ports_(fed_ports), in SymbolicShapeRefiner()2477 absl::flat_hash_map<string, absl::flat_hash_set<int>> fed_ports; in InferStatically() local2481 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()