| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 202 row_splits, argument
245 if not isinstance(row_splits, ops.Tensor):
248 row_splits.shape.assert_has_rank(1)
249 row_splits.set_shape([None])
252 self._row_splits = row_splits
351 row_splits = array_ops.concat([[0], math_ops.cumsum(row_lengths)], axis=0)
354 row_splits.set_shape([const_nrows + 1])
358 row_splits,
365 def from_row_splits(cls, values, row_splits, name=None): argument
397 if isinstance(row_splits, (list, tuple)) and not row_splits:
[all …]
|
| D | ragged_tensor_value.py | 36 def __init__(self, values, row_splits): argument
43 if not (isinstance(row_splits, (np.ndarray, np.generic)) and
44 row_splits.dtype == np.int64 and row_splits.ndim == 1):
49 self._row_splits = row_splits
51 row_splits = property( variable in RaggedTensorValue
72 rt_nested_splits = [self.row_splits]
75 rt_nested_splits.append(rt_values.row_splits)
|
| D | ragged_getitem.py | 139 nsplits = array_ops.shape(inner_rt.row_splits, out_type=dtypes.int64)[0]
152 starts = rt_input.row_splits[:-1]
153 limits = rt_input.row_splits[1:]
193 new_starts = rt_input.row_splits[:-1][row_key]
194 new_limits = rt_input.row_splits[1:][row_key]
248 nsplits = array_ops.shape(inner_rt.row_splits, out_type=dtypes.int64)[0]
265 inner_rt_starts = rt_input.row_splits[:-1]
266 inner_rt_limits = rt_input.row_splits[1:]
269 column_key.start, rt_input.row_splits[:-1], rt_input.row_splits[1:])
272 column_key.stop, rt_input.row_splits[:-1], rt_input.row_splits[1:])
|
| D | ragged_tensor_test.py | 127 values=[3, 1, 4, 1, 5, 9, 2, 6], row_splits=[0, 4, 4, 7, 8, 8])
133 rt1 = RaggedTensor.from_row_splits(values, row_splits=[0, 4, 4, 7, 8, 8])
145 values=[3, 1, 4, 1, 5, 9, 2, 6], row_splits=[0, 4, 4, 7, 8, 8])
147 values=inner_rt, row_splits=[0, 3, 3, 5])
164 values=array_ops.ones([5, 3]), row_splits=[0, 2, 5])
182 self.assertEqual(rt_value.row_splits.dtype, np.int64)
185 self.assertAllEqual(splits, rt_value.row_splits)
193 row_splits=splits2)
194 self.assertEqual(rt_value.row_splits.dtype, np.int64)
197 self.assertAllEqual(splits2, rt_value.row_splits)
[all …]
|
| D | ragged_to_sparse_op_test.py | 148 row_splits=[2, 3], values=[1, 2, 3])
154 row_splits=[0, 5], values=empty_vector)
156 row_splits=[0, 1],
158 row_splits=[0, 5], values=empty_vector))
166 row_splits=[0, 5],
168 row_splits=[0], values=empty_vector))
175 row_splits=empty_vector, values=[])
|
| D | ragged_factory_ops.py | 123 def _ragged_factory(values, row_splits): argument
124 row_splits = np.array(row_splits, dtype=np.int64)
125 return ragged_tensor_value.RaggedTensorValue(values, row_splits)
225 for row_splits in reversed(nested_splits):
226 values = ragged_factory(values, row_splits)
|
| D | ragged_batch_gather_ops.py | 88 checks = [check_ops.assert_equal(params.row_splits, indices.row_splits)]
91 batch_gather(params.values, indices.values), indices.row_splits)
111 indices.row_splits)
|
| D | ragged_math_ops.py | 199 segment_ids.row_splits,
200 data.row_splits,
209 data_row_lengths = data.row_splits[1:] - data.row_splits[:-1]
481 row_lengths = rt_input.row_splits[1:] - rt_input.row_splits[:-1]
488 num_segments = array_ops.shape(rt_input.row_splits)[0] - 1
490 rt_input.row_splits)
|
| D | ragged_string_ops.py | 92 input_splits=input_tensor.row_splits,
389 flat_result.char_values, flat_result.row_splits)
394 flat_result.char_to_byte_starts, flat_result.row_splits)
|
| D | ragged_array_ops.py | 150 splits.append(mask.row_splits)
185 segment_ids = segment_id_ops.row_splits_to_segment_ids(data.row_splits)
482 splits = input.row_splits
|
| D | ragged_concat_ops.py | 184 splits = [[rt_input.row_splits] for rt_input in rt_inputs]
271 concat_splits = permuted_rt.row_splits[::num_inputs]
|
| D | ragged_where_op.py | 150 selected_row_starts = array_ops.gather(condition.row_splits, selected_rows)
|
| D | ragged_test_util.py | 94 self._eval_tensor(tensor.row_splits))
|
| D | ragged_gather_ops.py | 238 flattened_index_tuples = array_ops.gather(params.row_splits,
|
| D | ragged_segment_op_test.py | 213 array_ops.placeholder_with_default(segment_ids.row_splits, None))
|
| D | ragged_dispatch_test.py | 446 array_ops.placeholder_with_default([1, 2, 3], shape=None), x.row_splits)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_UnicodeDecodeWithOffsets.pbtxt | 11 name: "row_splits"
73 The `row_splits` tensor indicates where the codepoints and start offsets for
77 `[row_splits[i]:row_splits[i+1]]`. Thus:
79 * `char_values[row_splits[i]+j]` is the Unicode codepoint for the `j`th
81 * `char_to_bytes_starts[row_splits[i]+j]` is the start byte offset for the `j`th
83 * `row_splits[i+1] - row_splits[i]` is the number of characters in the `i`th
|
| D | api_def_UnicodeDecode.pbtxt | 11 name: "row_splits"
64 The `row_splits` tensor indicates where the codepoints for
68 `[row_splits[i]:row_splits[i+1]]`. Thus:
70 * `char_values[row_splits[i]+j]` is the Unicode codepoint for the `j`th
72 * `row_splits[i+1] - row_splits[i]` is the number of characters in the `i`th
|
| D | api_def_RaggedGather.pbtxt | 41 contain this number of `row_splits` tensors. This value should equal
49 this number of `row_splits` tensors. This value should equal
|
| D | api_def_RaggedTensorToSparse.pbtxt | 6 description: "The `row_splits` for the `RaggedTensor`."
|
| D | api_def_RaggedRange.pbtxt | 18 description: "The `row_splits` for the returned `RaggedTensor`."
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.ragged.-ragged-tensor-value.pbtxt | 22 name: "row_splits"
35 … argspec: "args=[\'self\', \'values\', \'row_splits\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.-ragged-tensor.pbtxt | 23 name: "row_splits"
36 …argspec: "args=[\'self\', \'values\', \'row_splits\', \'cached_row_lengths\', \'cached_value_rowid…
68 …argspec: "args=[\'cls\', \'values\', \'row_splits\', \'name\'], varargs=None, keywords=None, defau…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 23 name: "row_splits"
36 …argspec: "args=[\'self\', \'values\', \'row_splits\', \'cached_row_lengths\', \'cached_value_rowid…
68 …argspec: "args=[\'cls\', \'values\', \'row_splits\', \'name\'], varargs=None, keywords=None, defau…
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | unicode_decode_op_test.py | 419 self.assertAllEqual(expected_row_splits, result.row_splits)
|