Searched refs:inputs_stacked (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/python/ops/parallel_for/ |
D | pfor.py | 366 inputs_stacked): argument 388 assert len(inputs_stacked) == len(self._enters) 410 for enter, inp, stacked in zip(self._enters, inputs, inputs_stacked): 414 wrapped_inp = wrap(inputs[i], inputs_stacked[i]) 519 inputs_stacked = [] 543 inputs_stacked.append(inp_stacked) 554 return init_values, inputs_stacked, shape_invariants 575 def _process_cond_stacked(self, conditions, indices, inputs, inputs_stacked, argument 587 for i, (inp, stacked) in enumerate(zip(inputs, inputs_stacked)): 606 def _process_body(self, pfor_input, inputs_stacked, new_indices, cond_stacked, argument [all …]
|