Searched refs:_unflatten_first_dim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1752 def _unflatten_first_dim(x, first_dim): function
1799 outputs = [_unflatten_first_dim(x, n) for x in outputs]
1828 output = _unflatten_first_dim(output, n)
1927 y = _unflatten_first_dim(y, n)
1952 outputs = [_unflatten_first_dim(x, n) for x in outputs[1:]]
1977 outputs = [_unflatten_first_dim(x, n) for x in outputs[1:]]
1998 outputs = [_unflatten_first_dim(x, n) for x in outputs]
2238 _unflatten_first_dim(output, pfor_input.pfor.loop_len_vector), True)
2781 output = _unflatten_first_dim(output, pfor_input.pfor.loop_len_vector)
2960 output = _unflatten_first_dim(output, [n])
[all …]
|