Searched refs:second_inputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 211 const StringCollection& second_inputs, in FuseFunctionNodes() argument
220 std::find(second_inputs.begin(), second_inputs.end(), parsed_name); in FuseFunctionNodes()
221 if (input_it == second_inputs.end()) continue; in FuseFunctionNodes()
223 auto arg_num = std::distance(second_inputs.begin(), input_it); in FuseFunctionNodes()
225 set_input(first_inputs, second_inputs, first_outputs, arg_num); in FuseFunctionNodes()
233 const StringCollection& second_inputs, in FuseReturns() argument
240 std::find(second_inputs.begin(), second_inputs.end(), return_input); in FuseReturns()
241 if (input_it == second_inputs.end()) continue; in FuseReturns()
243 auto input_idx = std::distance(second_inputs.begin(), input_it); in FuseReturns()
245 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,
|