Searched refs:num_non_control_inputs (Results 1 – 3 of 3) 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()
3654 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialAssocOpConstFolding() local3655 if (num_non_control_inputs <= 2) return false; in PartialAssocOpConstFolding()3656 const int num_control_inputs = node->input_size() - num_non_control_inputs; in PartialAssocOpConstFolding()3673 if (const_inputs_size == num_non_control_inputs && in PartialAssocOpConstFolding()3681 if (const_inputs_size > 1 && const_inputs_size < num_non_control_inputs && in PartialAssocOpConstFolding()3731 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialConcatConstFolding() local3732 if (num_non_control_inputs <= 3) return false; in PartialConcatConstFolding()3735 int end = num_non_control_inputs; in PartialConcatConstFolding()3740 end = num_non_control_inputs - 1; in PartialConcatConstFolding()3741 axis_arg = num_non_control_inputs - 1; in PartialConcatConstFolding()
3024 const int num_non_control_inputs = non_control_operands.size(); in matchAndRewrite() local3025 if (num_non_control_inputs <= 3) return failure(); in matchAndRewrite()3029 int end = num_non_control_inputs; in matchAndRewrite()3033 end = num_non_control_inputs - 1; in matchAndRewrite()3034 axis_arg = num_non_control_inputs - 1; in matchAndRewrite()