Searched refs:num_non_control_inputs (Results 1 – 2 of 2) sorted by relevance
281 NodeDef* node, int num_non_control_inputs, in RewriteIdentityNAndInputsOutputs() argument298 for (int i = 0; i < num_non_control_inputs; i++) { in RewriteIdentityNAndInputsOutputs()352 for (int i = 0; i < num_non_control_inputs; i++) { in RewriteIdentityNAndInputsOutputs()361 for (int i = num_non_control_inputs; i < num_inputs; i++) { in RewriteIdentityNAndInputsOutputs()385 const int num_non_control_inputs = NumNonControlInputs(*node); in SplitIdentityNInputs() local387 node->attr().at("T").list().type_size() != num_non_control_inputs || in SplitIdentityNInputs()388 terminal.second.size() >= num_non_control_inputs) { in SplitIdentityNInputs()393 node, num_non_control_inputs, terminal.second, graph, &node_map)); in SplitIdentityNInputs()
3079 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialAssocOpConstFolding() local3081 num_non_control_inputs > 2) { in PartialAssocOpConstFolding()3082 const int num_control_inputs = node->input_size() - num_non_control_inputs; in PartialAssocOpConstFolding()3098 if (const_inputs.size() == num_non_control_inputs && in PartialAssocOpConstFolding()3107 const_inputs.size() < num_non_control_inputs && in PartialAssocOpConstFolding()3155 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialConcatConstFolding() local3156 if (IsConcat(*node) && num_non_control_inputs > 3 && in PartialConcatConstFolding()3160 int end = num_non_control_inputs; in PartialConcatConstFolding()3165 end = num_non_control_inputs - 1; in PartialConcatConstFolding()3166 axis_arg = num_non_control_inputs - 1; in PartialConcatConstFolding()