Searched refs:first_inputs (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
D | basic_decoder_test.py | 78 (first_finished, first_inputs, first_state) = my_decoder.initialize() 81 constant_op.constant(0), first_inputs, first_state) 104 "first_inputs": first_inputs, 157 (first_finished, first_inputs, first_state) = my_decoder.initialize() 160 constant_op.constant(0), first_inputs, first_state) 178 "first_inputs": first_inputs, 233 (first_finished, first_inputs, first_state) = my_decoder.initialize() 236 constant_op.constant(0), first_inputs, first_state) 254 "first_inputs": first_inputs, 306 (first_finished, first_inputs, first_state) = my_decoder.initialize() [all …]
|
D | basic_decoder_v2_test.py | 73 first_inputs, 89 constant_op.constant(0), first_inputs, first_state) 112 "first_inputs": first_inputs, 149 (first_finished, first_inputs, first_state) = my_decoder.initialize( 166 constant_op.constant(0), first_inputs, first_state) 184 "first_inputs": first_inputs, 225 first_inputs, 242 constant_op.constant(0), first_inputs, first_state) 260 "first_inputs": first_inputs, 300 (first_finished, first_inputs, first_state) = my_decoder.initialize( [all …]
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | fusion_utils.cc | 210 void FuseFunctionNodes(const StringCollection& first_inputs, in FuseFunctionNodes() argument 225 set_input(first_inputs, second_inputs, first_outputs, arg_num); in FuseFunctionNodes() 232 void FuseReturns(const StringCollection& first_inputs, in FuseReturns() argument 245 set_input(first_inputs, second_inputs, first_outputs, input_idx); in FuseReturns() 313 string ComposeInput(const StringCollection& first_inputs, in ComposeInput() argument 354 string SameInput(const StringCollection& first_inputs, in SameInput() argument 357 return first_inputs.at(arg_num); in SameInput() 461 const auto first_inputs = GetFunctionInputs(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 | 72 string ComposeInput(const StringCollection& first_inputs, 104 string SameInput(const StringCollection& first_inputs,
|