Searched refs:first_outputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 213 const StringCollection& first_outputs, in FuseFunctionNodes() argument
226 set_input(first_inputs, second_inputs, first_outputs, arg_num); in FuseFunctionNodes()
235 const StringCollection& first_outputs, in FuseReturns() argument
246 set_input(first_inputs, second_inputs, first_outputs, input_idx); in FuseReturns()
316 const StringCollection& first_outputs, int arg_num) { in ComposeInput() argument
318 return first_outputs.at(arg_num); in ComposeInput()
357 const StringCollection& first_outputs, int arg_num) { in SameInput() argument
474 const auto first_outputs = GetFunctionOutputs(first_function); in FuseFunctions() local
475 FuseFunctionNodes(first_inputs, second_inputs, first_outputs, set_input, in FuseFunctions()
477 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);
|