Searched refs:num_non_control_inputs (Results 1 – 2 of 2) sorted by relevance
299 NodeDef* node, int num_non_control_inputs, in RewriteIdentityNAndInputsOutputs() argument316 for (int i = 0; i < num_non_control_inputs; i++) { in RewriteIdentityNAndInputsOutputs()370 for (int i = 0; i < num_non_control_inputs; i++) { in RewriteIdentityNAndInputsOutputs()379 for (int i = num_non_control_inputs; i < num_inputs; i++) { in RewriteIdentityNAndInputsOutputs()403 const int num_non_control_inputs = NumNonControlInputs(*node); in SplitIdentityNInputs() local406 node->attr().at("T").list().type_size() != num_non_control_inputs || in SplitIdentityNInputs()407 terminal_second_size >= num_non_control_inputs) { in SplitIdentityNInputs()412 node, num_non_control_inputs, terminal.second, graph, &node_map)); in SplitIdentityNInputs()
3595 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialAssocOpConstFolding() local3596 if (num_non_control_inputs <= 2) return false; in PartialAssocOpConstFolding()3597 const int num_control_inputs = node->input_size() - num_non_control_inputs; in PartialAssocOpConstFolding()3614 if (const_inputs_size == num_non_control_inputs && in PartialAssocOpConstFolding()3622 if (const_inputs_size > 1 && const_inputs_size < num_non_control_inputs && in PartialAssocOpConstFolding()3672 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialConcatConstFolding() local3673 if (num_non_control_inputs <= 3) return false; in PartialConcatConstFolding()3676 int end = num_non_control_inputs; in PartialConcatConstFolding()3681 end = num_non_control_inputs - 1; in PartialConcatConstFolding()3682 axis_arg = num_non_control_inputs - 1; in PartialConcatConstFolding()