Searched refs:_unflatten_first_dim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1734 def _unflatten_first_dim(x, first_dim): function
1781 outputs = [_unflatten_first_dim(x, n) for x in outputs]
1810 output = _unflatten_first_dim(output, n)
1909 y = _unflatten_first_dim(y, n)
1934 outputs = [_unflatten_first_dim(x, n) for x in outputs[1:]]
1959 outputs = [_unflatten_first_dim(x, n) for x in outputs[1:]]
1980 outputs = [_unflatten_first_dim(x, n) for x in outputs]
2130 _unflatten_first_dim(output, pfor_input.pfor.loop_len_vector), True)
2576 output = _unflatten_first_dim(output, pfor_input.pfor.loop_len_vector)
2755 output = _unflatten_first_dim(output, [n])
[all …]
|