Searched refs:second_inputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 212 const StringCollection& second_inputs, in FuseFunctionNodes() argument
221 std::find(second_inputs.begin(), second_inputs.end(), parsed_name); in FuseFunctionNodes()
222 if (input_it == second_inputs.end()) continue; in FuseFunctionNodes()
224 auto arg_num = std::distance(second_inputs.begin(), input_it); in FuseFunctionNodes()
226 set_input(first_inputs, second_inputs, first_outputs, arg_num); in FuseFunctionNodes()
234 const StringCollection& second_inputs, in FuseReturns() argument
241 std::find(second_inputs.begin(), second_inputs.end(), return_input); in FuseReturns()
242 if (input_it == second_inputs.end()) continue; in FuseReturns()
244 auto input_idx = std::distance(second_inputs.begin(), input_it); in FuseReturns()
246 set_input(first_inputs, second_inputs, first_outputs, input_idx); in FuseReturns()
[all …]
|
| D | fusion_utils.h | 73 const StringCollection& second_inputs,
105 const StringCollection& second_inputs,
|