Searched refs:RowPartitionSpec (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | row_partition_test.py | 33 from tensorflow.python.ops.ragged.row_partition import RowPartitionSpec
724 spec=RowPartitionSpec(nrows=None, nvals=None, dtype=dtypes.int64)),
728 spec=RowPartitionSpec(nrows=2, nvals=None, dtype=dtypes.int64)),
732 spec=RowPartitionSpec(nrows=None, nvals=8, dtype=dtypes.int64))
745 spec=RowPartitionSpec(nrows=2, nvals=8, dtype=dtypes.int64),
750 spec=RowPartitionSpec(nrows=None, nvals=8, dtype=dtypes.int64),
926 spec = RowPartitionSpec()
945 spec = RowPartitionSpec(nrows, nvals, uniform_row_length, dtype)
967 RowPartitionSpec(nrows, nvals, uniform_row_length, dtype)
970 RowPartitionSpec(),
[all …]
|
| D | dynamic_ragged_shape_test.py | 47 from tensorflow.python.ops.ragged.row_partition import RowPartitionSpec
2116 RowPartitionSpec(nrows=2,
2130 RowPartitionSpec(nrows=2,
3093 a: RowPartitionSpec, argument
3094 b: RowPartitionSpec, argument
3127 RowPartitionSpec(
3132 RowPartitionSpec(
3174 RowPartitionSpec(
3179 RowPartitionSpec(
3184 RowPartitionSpec(
[all …]
|
| D | dynamic_ragged_shape.py | 39 from tensorflow.python.ops.ragged.row_partition import RowPartitionSpec
68 …new_head = RowPartitionSpec(uniform_row_length=spec._dimension(0), # pylint:disable=protected-acc…
94 RowPartitionSpec(
103 result.append(RowPartitionSpec(nrows=nrows,
109 RowPartitionSpec(
1117 def __init__(self, row_partitions: Tuple[RowPartitionSpec, ...], argument
1138 if not all(isinstance(rp, RowPartitionSpec) for rp in row_partitions):
1195 RowPartitionSpec( # pylint: disable=g-complex-comprehension
1266 RowPartitionSpec(dtype=dtype) for _ in range(num_row_partitions)
1292 rp_specs.append(RowPartitionSpec(
[all …]
|
| D | row_partition.py | 1198 return RowPartitionSpec.from_value(self)
1209 class RowPartitionSpec(type_spec.TypeSpec): class
1277 if not super(RowPartitionSpec, self).is_compatible_with(other):
1358 if not RowPartitionSpec._dimensions_compatible(nrows, nvals, ncols):
1365 return RowPartitionSpec(nrows=nrows[0],
1373 return RowPartitionSpec(nrows, nvals, self._uniform_row_length, dtype)
1383 return RowPartitionSpec(nrows, nvals, uniform_row_length, dtype)
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_spec_test.py | 50 PARTITION_SPEC = row_partition.RowPartitionSpec()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | struct.proto | 138 ROW_PARTITION_SPEC = 10; // RowPartitionSpec from ragged/row_partition.py
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | nested_structure_coder.py | 486 row_partition.RowPartitionSpec,
|