Searched refs:row_partition (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 45 from tensorflow.python.ops.ragged.row_partition import RowPartition
248 def __init__(self, values, row_partition, internal=False): argument
282 if not isinstance(row_partition, RowPartition):
284 row_partition)
290 assert row_partition.dtype == values._row_partition.dtype
293 self._row_partition = row_partition
300 def _from_row_partition(cls, values, row_partition, validate=True): argument
320 if not isinstance(row_partition, RowPartition):
324 values, row_partition = cls._convert_values_and_partition(
325 values, row_partition, "partition")
[all …]
|
| D | row_partition_test.py | 32 from tensorflow.python.ops.ragged import row_partition
33 from tensorflow.python.ops.ragged.row_partition import RowPartition
34 from tensorflow.python.ops.ragged.row_partition import RowPartitionSpec
74 internal=row_partition._row_partition_factory_key)
88 internal=row_partition._row_partition_factory_key)
93 internal=row_partition._row_partition_factory_key)
100 internal=row_partition._row_partition_factory_key)
|
| D | BUILD | 295 name = "row_partition",
296 srcs = ["row_partition.py"],
324 ":row_partition",
568 ":row_partition",
|
| D | ragged_tensor_test.py | 50 from tensorflow.python.ops.ragged.row_partition import RowPartition
153 rt = RaggedTensor(values=values, row_partition=rp, internal=True)
165 RaggedTensor(values=values, row_partition=rp)
170 RaggedTensor(values=range(7), row_partition=rp, internal=True)
175 values=values, row_partition=[0, 2, 2, 5, 6, 7], internal=True)
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_test.py | 38 from tensorflow.python.ops.ragged import row_partition
216 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]))
[all …]
|
| D | structured_tensor.py | 40 from tensorflow.python.ops.ragged.row_partition import RowPartition
951 def partition_outer_dimension(self, row_partition): argument
974 if not isinstance(row_partition, RowPartition):
978 return _partition_outer_dimension(self, row_partition)
1443 row_partition=new_partitions[0])
1458 def _partition_outer_dimension(value, row_partition): argument
1483 is_ragged = row_partition.uniform_row_length() is None
1486 [[row_partition.nrows(),
1487 row_partition.uniform_row_length()],
1488 array_ops.shape(value, out_type=row_partition.dtype)[1:]],
[all …]
|
| D | BUILD | 48 "//tensorflow/python/ops/ragged:row_partition",
75 "//tensorflow/python/ops/ragged:row_partition",
100 "//tensorflow/python/ops/ragged:row_partition",
|
| D | structured_array_ops.py | 24 from tensorflow.python.ops.ragged.row_partition import RowPartition
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | nested_structure_coder.py | 52 from tensorflow.python.ops.ragged import row_partition
518 row_partition.RowPartitionSpec,
|
| D | BUILD | 586 "//tensorflow/python/ops/ragged:row_partition",
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor.pbtxt | 41 …argspec: "args=[\'self\', \'values\', \'row_partition\', \'internal\'], varargs=None, keywords=Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 41 …argspec: "args=[\'self\', \'values\', \'row_partition\', \'internal\'], varargs=None, keywords=Non…
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | struct.proto | 138 ROW_PARTITION_SPEC = 10; // RowPartitionSpec from ragged/row_partition.py
|