Home
last modified time | relevance | path

Searched refs:row_limits (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition.py393 row_limits = math_ops.cumsum(row_lengths)
394 row_splits = array_ops.concat([[0], row_limits], axis=0)
444 def from_row_limits(cls, row_limits, validate=True, preferred_dtype=None): argument
462 with ops.name_scope(None, "RowPartitionFromRowLimits", [row_limits]):
463 row_limits = cls._convert_row_partition(row_limits, "row_limits",
465 row_limits.shape.assert_has_rank(1)
470 check_ops.assert_rank(row_limits, 1, message=msg),
471 check_ops.assert_non_negative(row_limits[:1], message=msg),
472 _assert_monotonic_increasing(row_limits, message=msg),
474 row_limits = control_flow_ops.with_dependencies(checks, row_limits)
[all …]
Dragged_tensor_supported_values_test.py171 row_limits = constant_op.constant([2, 2, 5, 6, 8], dtypes.int64)
172 rt = RaggedTensor.from_row_limits(values, row_limits)
175 self.assertAllEqual(rt.row_limits(), row_limits)
Drow_partition_test.py56 rt5 = RowPartition.from_row_limits(row_limits=[4, 4, 7, 8, 8])
223 row_limits = constant_op.constant([2, 2, 5, 6, 7], dtypes.int64)
225 rp = RowPartition.from_row_limits(row_limits, validate=False)
228 rp_row_limits = rp.row_limits()
233 self.assertAllEqual(rp_row_limits, row_limits)
Dragged_tensor_test.py82 rt5 = RaggedTensor.from_row_limits(values, row_limits=[4, 4, 7, 8, 8])
345 row_limits = constant_op.constant([2, 2, 5, 6, 7], dtypes.int64)
347 rt = RaggedTensor.from_row_limits(values, row_limits, validate=False)
353 rt_row_limits = rt.row_limits()
358 self.assertAllEqual(rt_row_limits, row_limits)
668 self.assertAllEqual(rt.row_limits(), [2, 2, 5, 6, 7])
695 self.assertAllEqual(rt.row_limits(), [2, 2, 5, 6, 7])
731 self.assertAllEqual(rt.row_limits(), [2, 3, 3, 5])
977 'test': RaggedTensor.row_limits,
Dragged_tensor.py537 def from_row_limits(cls, values, row_limits, name=None, validate=True): argument
565 with ops.name_scope(name, "RaggedFromRowLimits", [values, row_limits]):
567 row_limits=row_limits,
1151 def row_limits(self, name=None): member in RaggedTensor
1174 return self._row_partition.row_limits()
/external/tensorflow/tensorflow/python/keras/engine/
Dragged_keras_tensor_test.py243 row_limits = constant_op.constant([2, 2, 5, 6, 7], dtypes.int64)
247 inp, row_limits, validate=False)
252 x, row_limits, validate=False)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor.pbtxt69 …argspec: "args=[\'cls\', \'values\', \'row_limits\', \'name\', \'validate\'], varargs=None, keywor…
124 name: "row_limits"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor.pbtxt69 …argspec: "args=[\'cls\', \'values\', \'row_limits\', \'name\', \'validate\'], varargs=None, keywor…
124 name: "row_limits"