Searched refs:num_data_inputs (Results 1 – 2 of 2) sorted by relevance
822 const int num_data_inputs = dead.node->attr().at("N").i(); in RemoveDeadBranches() local823 if (num_data_inputs > 2) { in RemoveDeadBranches()846 if (local_dead_merge_inputs[dead.node].size() == num_data_inputs) { in RemoveDeadBranches()904 const int num_data_inputs = merge_node->attr().at("N").i(); in RemoveDeadBranches() local905 if (merge_node->input_size() != num_data_inputs) { in RemoveDeadBranches()910 } else if (dead_inputs.size() != 1 || num_data_inputs != 2) { in RemoveDeadBranches()915 << num_data_inputs; in RemoveDeadBranches()
3482 int num_data_inputs = orig_node->in_edges().size(); in RewriteNodeForLayoutPropagation() local3486 num_data_inputs--; in RewriteNodeForLayoutPropagation()3491 gtl::InlinedVector<std::pair<Node*, int>, 4> inputs(num_data_inputs); in RewriteNodeForLayoutPropagation()3566 int num_data_inputs = orig_node->in_edges().size(); in RewriteNodeForJustOpNameChange() local3570 num_data_inputs--; in RewriteNodeForJustOpNameChange()3574 gtl::InlinedVector<std::pair<Node*, int>, 4> inputs(num_data_inputs); in RewriteNodeForJustOpNameChange()3957 int num_data_inputs = n->num_inputs() / 2; in FixMklMetaDataEdges() local3958 for (int idx = 0; idx < num_data_inputs; idx++) { in FixMklMetaDataEdges()