Searched refs:user_defined_nodes (Results 1 – 1 of 1) sorted by relevance
181 std::unordered_map<string, const NodeDef*> user_defined_nodes; in InferShapesForFunction() local183 user_defined_nodes[node_def.name()] = &node_def; in InferShapesForFunction()190 if (user_defined_nodes.find(node_name) != user_defined_nodes.end()) { in InferShapesForFunction()197 user_defined_nodes[node_name]; in InferShapesForFunction()