Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition_test.py33 from tensorflow.python.ops.ragged.row_partition import RowPartition
46 rp = RowPartition.from_row_splits(row_splits=[0, 4, 4, 7, 8, 8])
51 rt1 = RowPartition.from_row_splits(row_splits=[0, 4, 4, 7, 8, 8])
52 rt2 = RowPartition.from_row_lengths(row_lengths=[4, 0, 3, 1, 0])
53 rt3 = RowPartition.from_value_rowids(
55 rt4 = RowPartition.from_row_starts(row_starts=[0, 4, 4, 7, 8], nvals=8)
56 rt5 = RowPartition.from_row_limits(row_limits=[4, 4, 7, 8, 8])
62 inner_rt = RowPartition.from_row_splits(row_splits=[0, 4, 4, 7, 8, 8])
63 outer_rt = RowPartition.from_row_splits(row_splits=[0, 3, 3, 5])
72 rp = RowPartition(
[all …]
Drow_partition.py51 class RowPartition(composite_tensor.CompositeTensor): class
636 return RowPartition(
853 return RowPartition(
906 return RowPartition(
916 return RowPartition(
926 return RowPartition(
936 return RowPartition(
1005 return RowPartition(
1034 value_type = property(lambda self: RowPartition)
1141 return RowPartition.from_row_splits(tensor, validate=False)
[all …]
Dragged_tensor.py45 from tensorflow.python.ops.ragged.row_partition import RowPartition
52 _convert_row_partition = RowPartition._convert_row_partition
282 if not isinstance(row_partition, RowPartition):
320 if not isinstance(row_partition, RowPartition):
398 row_partition = RowPartition.from_value_rowids(
446 row_partition = RowPartition.from_row_splits(
489 row_partition = RowPartition.from_row_lengths(
528 row_partition = RowPartition.from_row_starts(
566 row_partition = RowPartition.from_row_limits(
639 row_partition = RowPartition.from_uniform_row_length(
[all …]
Dragged_tensor_test.py50 from tensorflow.python.ops.ragged.row_partition import RowPartition
152 rp = RowPartition.from_row_splits(row_splits)
161 rp = RowPartition.from_row_splits(row_splits)
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_array_ops.py24 from tensorflow.python.ops.ragged.row_partition import RowPartition
145 new_partition = RowPartition.from_uniform_row_length(
151 return st.row_partitions + (RowPartition.from_uniform_row_length(
156 return st.row_partitions[:axis - 1] + (RowPartition.from_uniform_row_length(
Dstructured_tensor_test.py216 lambda: [row_partition.RowPartition.from_row_lengths([3, 7])],
270 row_partition.RowPartition.from_row_lengths([3]),
271 row_partition.RowPartition.from_row_lengths([2, 0, 1]),
272 row_partition.RowPartition.from_uniform_row_length(3, nvals=9)
415 [row_partition.RowPartition.from_row_lengths([3, 4])],
514 row_partition.RowPartition.from_row_splits([0, 2, 2, 3, 4]))
518 row_partition.RowPartition.from_row_lengths([1, 0, 3, 0]))
524 row_partition.RowPartition.from_uniform_row_length(
544 row_partition.RowPartition.from_row_splits([0, 1, 2]))
546 row_partition.RowPartition.from_row_splits([0, 1, 2]))
[all …]
Dstructured_tensor.py40 from tensorflow.python.ops.ragged.row_partition import RowPartition
974 if not isinstance(row_partition, RowPartition):
1288 RowPartition.from_uniform_row_length(
1545 (RowPartition.from_row_splits(merged_splits),) +