Searched refs:value_rowids (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_test.py | 136 values, value_rowids=[0, 0, 0, 0, 2, 2, 2, 3], nrows=5)
256 value_rowids = constant_op.constant([0, 0, 2, 2, 2, 3, 4], dtypes.int64)
258 rt = RaggedTensor.from_value_rowids(values, value_rowids)
264 rt_value_rowids = rt.value_rowids()
268 self.assertIs(rt_value_rowids, value_rowids) # cached_value_rowids
269 self.assertAllEqual(rt_value_rowids, value_rowids)
278 value_rowids = constant_op.constant([0, 0, 2, 2, 2, 3, 4], dtypes.int64)
279 value_rowids = array_ops.placeholder_with_default(value_rowids, shape=None)
281 rt = RaggedTensor.from_value_rowids(values, value_rowids)
290 rt_value_rowids = rt.value_rowids()
[all …]
|
| D | ragged_tensor.py | 269 def from_value_rowids(cls, values, value_rowids, nrows=None, name=None): argument
310 [values, value_rowids, nrows]):
312 value_rowids = ops.convert_to_tensor(
313 value_rowids, dtypes.int64, name="value_rowids")
315 const_rowids = tensor_util.constant_value(value_rowids)
317 nrows = array_ops.concat([value_rowids[-1:], [-1]], axis=0)[0] + 1
328 const_rowids = tensor_util.constant_value(value_rowids)
335 value_rowids.shape.assert_has_rank(1)
337 values.shape[:1].assert_is_compatible_with(value_rowids.shape)
344 value_rowids_int32 = math_ops.cast(value_rowids, dtypes.int32)
[all …]
|
| D | ragged_where_op.py | 149 selected_rows = array_ops.gather(condition.value_rowids(), first_index)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 84 …argspec: "args=[\'cls\', \'values\', \'value_rowids\', \'nrows\', \'name\'], varargs=None, keyword…
119 name: "value_rowids"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor.pbtxt | 84 …argspec: "args=[\'cls\', \'values\', \'value_rowids\', \'nrows\', \'name\'], varargs=None, keyword…
119 name: "value_rowids"
|