Searched refs:_flatten_array_with_offset (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 2909 def _flatten_array_with_offset(ids, offset_delta, num_rows): function
2949 indices = _flatten_array_with_offset(indices, array_ops.shape(data)[1], n)
2953 segment_ids = _flatten_array_with_offset(segment_ids, num_segments, n)
2976 indices = _flatten_array_with_offset(indices, dim0, n)
2978 segment_ids = _flatten_array_with_offset(segment_ids, num_segments, n)
|