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"
|