Home
last modified time | relevance | path

Searched refs:stack_indices (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py794 def stack_inputs(self, stack_indices=None, tile_variants=False): argument
805 if stack_indices is None:
806 stack_indices = range(len(self._inputs))
808 for i in stack_indices:
1745 pfor_input.stack_inputs(stack_indices=input_indices)
1986 pfor_input.stack_inputs(stack_indices=[2])
2397 pfor_input.stack_inputs(stack_indices=[1])
2410 pfor_input.stack_inputs(stack_indices=[1])
2420 pfor_input.stack_inputs(stack_indices=range(n - 1))
4599 stack_indices=indices_to_stack, tile_variants=True)