Searched refs:row_limits (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 439 row_limits = math_ops.cumsum(row_lengths)
440 row_splits = array_ops.concat([[0], row_limits], axis=0)
481 def from_row_limits(cls, values, row_limits, name=None): argument
504 with ops.name_scope(name, "RaggedFromRowLimits", [values, row_limits]):
506 row_limits = ops.convert_to_tensor(row_limits, dtypes.int64, "row_limits")
507 row_limits.shape.assert_has_rank(1)
509 row_splits = array_ops.concat([zero, row_limits], axis=0)
856 def row_limits(self, name=None): member in RaggedTensor
|
| D | ragged_tensor_test.py | 138 rt5 = RaggedTensor.from_row_limits(values, row_limits=[4, 4, 7, 8, 8])
403 row_limits = constant_op.constant([2, 2, 5, 6, 7], dtypes.int64)
405 rt = RaggedTensor.from_row_limits(values, row_limits)
411 rt_row_limits = rt.row_limits()
416 self.assertAllEqual(rt_row_limits, row_limits)
617 self.assertAllEqual(rt.row_limits(), [2, 2, 5, 6, 7])
645 self.assertEqual(self.eval_to_list(rt.row_limits()), [2, 2, 5, 6, 7])
678 self.assertEqual(self.eval_to_list(rt.row_limits()), [2, 3, 3, 5])
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor.pbtxt | 64 …argspec: "args=[\'cls\', \'values\', \'row_limits\', \'name\'], varargs=None, keywords=None, defau…
99 name: "row_limits"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 64 …argspec: "args=[\'cls\', \'values\', \'row_limits\', \'name\'], varargs=None, keywords=None, defau…
99 name: "row_limits"
|