Searched refs:row_partitions (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 111 def __init__(self, fields, shape, nrows, row_partitions, internal=False): argument
134 assert isinstance(row_partitions, tuple), row_partitions
138 self._row_partitions = row_partitions
145 row_partitions=None, argument
198 if rank < 2 and row_partitions:
202 if row_partitions is not None:
203 row_partitions = tuple(row_partitions)
204 if len(row_partitions) != max(0, rank - 1):
207 row_partitions = ()
221 shape_dtype = _find_shape_dtype(fields, nrows, row_partitions)
[all …]
|
| D | structured_array_ops.py | 135 row_partitions=new_row_partitions,
147 return (new_partition,) + st.row_partitions
150 st.row_partitions[axis - 2].nvals() if (axis - 2 >= 0) else st.nrows())
151 return st.row_partitions + (RowPartition.from_uniform_row_length(
155 st.row_partitions[axis - 1].nrows() if (axis - 1 >= 0) else st.nrows())
156 return st.row_partitions[:axis - 1] + (RowPartition.from_uniform_row_length(
157 1, nvals, nrows=nvals, validate=False),) + st.row_partitions[axis - 1:]
|
| D | structured_tensor_test.py | 303 row_partitions=None): argument
308 if callable(row_partitions):
309 row_partitions = row_partitions() # deferred construction.
315 row_partitions=row_partitions,
390 row_partitions=[0],
396 row_partitions=[],
414 row_partitions=lambda:
447 row_partitions=None, argument
458 if callable(row_partitions):
459 row_partitions = row_partitions() # deferred construction.
[all …]
|