Searched refs:original_input_node (Results 1 – 1 of 1) sorted by relevance
375 Node* original_input_node = FindMutableNodeByName(node_name, graph); in TransformGraphToAddAggregatedInputNode() local376 CHECK_NOTNULL(original_input_node); in TransformGraphToAddAggregatedInputNode()377 CHECK_EQ(1, original_input_node->num_outputs()); // replaced by identity. in TransformGraphToAddAggregatedInputNode()386 original_input_node->attrs(), &data_types, &shapes); in TransformGraphToAddAggregatedInputNode()393 for (const Edge* out_edge : original_input_node->out_edges()) { in TransformGraphToAddAggregatedInputNode()399 original_input_nodes[i] = original_input_node; in TransformGraphToAddAggregatedInputNode()414 for (Node* original_input_node : original_input_nodes) { in TransformGraphToAddAggregatedInputNode()415 graph->RemoveNode(original_input_node); in TransformGraphToAddAggregatedInputNode()