Searched refs:first_outputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 212 const StringCollection& first_outputs, in FuseFunctionNodes() argument
225 set_input(first_inputs, second_inputs, first_outputs, arg_num); in FuseFunctionNodes()
234 const StringCollection& first_outputs, in FuseReturns() argument
245 set_input(first_inputs, second_inputs, first_outputs, input_idx); in FuseReturns()
315 const StringCollection& first_outputs, int arg_num) { in ComposeInput() argument
317 return first_outputs.at(arg_num); in ComposeInput()
356 const StringCollection& first_outputs, int arg_num) { in SameInput() argument
463 const auto first_outputs = GetFunctionOutputs(first_function); in FuseFunctions() local
464 FuseFunctionNodes(first_inputs, second_inputs, first_outputs, set_input, in FuseFunctions()
466 FuseReturns(first_inputs, second_inputs, first_outputs, set_input, in FuseFunctions()
|
| D | fusion_utils.h | 74 const StringCollection& first_outputs, int arg_num);
106 const StringCollection& first_outputs, int arg_num);
|