Searched refs:row_lengths (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | segment_id_ops.py | 60 row_lengths = splits[1:] - splits[:-1]
63 return ragged_util.repeat(indices, repeats=row_lengths, axis=0)
97 row_lengths = math_ops.bincount(
102 splits = array_ops.concat([[0], math_ops.cumsum(row_lengths)], axis=0)
|
| D | ragged_tensor.py | 346 row_lengths = math_ops.bincount(
351 row_splits = array_ops.concat([[0], math_ops.cumsum(row_lengths)], axis=0)
353 row_lengths.set_shape([const_nrows])
359 cached_row_lengths=row_lengths,
406 def from_row_lengths(cls, values, row_lengths, name=None): argument
434 with ops.name_scope(name, "RaggedFromRowLengths", [values, row_lengths]):
436 row_lengths = ops.convert_to_tensor(row_lengths, dtypes.int64,
438 row_lengths.shape.assert_has_rank(1)
439 row_limits = math_ops.cumsum(row_lengths)
444 cached_row_lengths=row_lengths,
[all …]
|
| D | ragged_batch_gather_with_default_op.py | 111 row_lengths = ragged_array_ops.expand_dims(
112 params.row_lengths(axis=num_batch_dimensions),
114 upper_bounds = math_ops.cast(row_lengths, indices.dtype)
|
| D | ragged_row_lengths_op_test.py | 122 lengths = rt.row_lengths(axis)
143 rt.row_lengths(axis)
|
| D | ragged_math_ops.py | 481 row_lengths = rt_input.row_splits[1:] - rt_input.row_splits[:-1]
482 num_segments = math_ops.maximum(math_ops.reduce_max(row_lengths), 0)
483 segment_ids = range(row_lengths).values
|
| D | ragged_tensor_test.py | 134 rt2 = RaggedTensor.from_row_lengths(values, row_lengths=[4, 0, 3, 1, 0])
423 row_lengths = constant_op.constant([2, 0, 3, 1, 1], dtypes.int64)
425 rt = RaggedTensor.from_row_lengths(values, row_lengths)
431 rt_row_lengths = rt.row_lengths()
435 self.assertIs(rt_row_lengths, row_lengths) # cached_nrows
437 self.assertAllEqual(rt_row_lengths, row_lengths)
618 self.assertAllEqual(rt.row_lengths(), [2, 0, 3, 1, 1])
646 self.assertEqual(self.eval_to_list(rt.row_lengths()), [2, 0, 3, 1, 1])
679 self.assertEqual(self.eval_to_list(rt.row_lengths()), [2, 1, 0, 2])
|
| D | ragged_batch_gather_ops.py | 104 num_indices = indices.row_lengths()
|
| D | ragged_array_ops.py | 177 lengths = data.row_lengths()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 60 …argspec: "args=[\'cls\', \'values\', \'row_lengths\', \'name\'], varargs=None, keywords=None, defa…
95 name: "row_lengths"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor.pbtxt | 60 …argspec: "args=[\'cls\', \'values\', \'row_lengths\', \'name\'], varargs=None, keywords=None, defa…
95 name: "row_lengths"
|