Home
last modified time | relevance | path

Searched refs:value_rowids (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition.py115 value_rowids=None, argument
166 for tensor in [row_lengths, value_rowids, nrows]:
173 self._value_rowids = value_rowids
189 value_rowids,
237 [value_rowids, nrows]):
238 value_rowids = cls._convert_row_partition(value_rowids, "value_rowids",
241 const_rowids = tensor_util.constant_value(value_rowids)
243 nrows = array_ops.concat([value_rowids[-1:], [-1]], axis=0)[0] + 1
248 const_nrows, value_rowids.dtype, name="nrows")
250 nrows = ops.convert_to_tensor(nrows, value_rowids.dtype, "nrows")
[all …]
Drow_partition_test.py54 value_rowids=[0, 0, 0, 0, 2, 2, 2, 3], nrows=5)
108 value_rowids = constant_op.constant([0, 0, 2, 2, 2, 3, 4], dtypes.int64)
110 rp = RowPartition.from_value_rowids(value_rowids, validate=False)
114 rp_value_rowids = rp.value_rowids()
117 self.assertIs(rp_value_rowids, value_rowids) # value_rowids
118 self.assertAllEqual(rp_value_rowids, value_rowids)
124 value_rowids = constant_op.constant([0, 0, 2, 2, 2, 3, 4], dtypes.int64)
125 value_rowids = array_ops.placeholder_with_default(value_rowids, shape=None)
127 rp = RowPartition.from_value_rowids(value_rowids, validate=False)
129 rp_value_rowids = rp.value_rowids()
[all …]
Dragged_tensor_test.py80 values, value_rowids=[0, 0, 0, 0, 2, 2, 2, 3], nrows=5)
184 value_rowids = constant_op.constant([0, 0, 2, 2, 2, 3, 4], dtypes.int64)
186 rt = RaggedTensor.from_value_rowids(values, value_rowids, validate=False)
192 rt_value_rowids = rt.value_rowids()
196 self.assertIs(rt_value_rowids, value_rowids) # cached_value_rowids
197 self.assertAllEqual(rt_value_rowids, value_rowids)
205 value_rowids = constant_op.constant([0, 0, 2, 2, 2, 3, 4], dtypes.int64)
206 value_rowids = array_ops.placeholder_with_default(value_rowids, shape=None)
208 rt = RaggedTensor.from_value_rowids(values, value_rowids, validate=False)
217 rt_value_rowids = rt.value_rowids()
[all …]
Dragged_tensor.py327 value_rowids_shape = row_partition.value_rowids().shape
350 value_rowids,
397 [values, value_rowids, nrows]):
399 value_rowids=value_rowids,
698 for value_rowids, nrows in reversed(
701 result, value_rowids, nrows, validate=validate)
1039 def value_rowids(self, name=None): member in RaggedTensor
1064 return self._row_partition.value_rowids()
1096 rt_nested_ids = [self.value_rowids()]
1099 rt_nested_ids.append(rt_values.value_rowids())
[all …]
Dragged_array_ops.py610 value_rowids=array_ops.stack([partitions]),
619 value_rowids = array_ops.gather(partitions, permutation)
622 value_rowids[-1:],
627 values, value_rowids, nrows=num_partitions, validate=False)
Dragged_to_tensor_op_test.py73 rowids = rt.value_rowids()
449 value_rowids=constant_op.constant([0, 1, 2, 3], dtype=dtypes.int64),
497 value_rowids=constant_op.constant([], dtype=dtypes.int64),
687 rt.value_rowids(), rt.nrows())
Dragged_string_ops.py85 values=values, value_rowids=indices[:, 0], nrows=shape[0],
524 value_rowids=sparse_result.indices[:, 0],
588 value_rowids=sparse_result.indices[:, 0],
Dragged_where_op.py236 selected_rows = array_ops.gather(condition.value_rowids(), first_index)
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_tensor_op.cc121 const RowPartitionTensor& value_rowids) { in GetMaxWidthValueRowID() argument
122 const INDEX_TYPE index_length = value_rowids.size(); in GetMaxWidthValueRowID()
127 INDEX_TYPE first_equal_index_value = value_rowids(0); in GetMaxWidthValueRowID()
130 const INDEX_TYPE value = value_rowids(i); in GetMaxWidthValueRowID()
262 const RowPartitionTensor& value_rowids, in CalculateOutputIndexValueRowID() argument
266 const INDEX_TYPE index_size = value_rowids.size(); in CalculateOutputIndexValueRowID()
273 INDEX_TYPE current_value_rowid = value_rowids(0); in CalculateOutputIndexValueRowID()
278 INDEX_TYPE next_value_rowid = value_rowids(i); in CalculateOutputIndexValueRowID()
296 DCHECK_EQ(result->size(), value_rowids.size()); in CalculateOutputIndexValueRowID()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc161 auto value_rowids = [tensor](int index) { in GetMaxWidthValueRowID() local
173 int first_equal_index_value = value_rowids(0); in GetMaxWidthValueRowID()
176 const int value = value_rowids(i); in GetMaxWidthValueRowID()
331 void CalculateOutputIndexValueRowID(const TfLiteTensor& value_rowids, in CalculateOutputIndexValueRowID() argument
335 const RuntimeShape tensor_shape(value_rowids.dims->size, in CalculateOutputIndexValueRowID()
336 value_rowids.dims->data); in CalculateOutputIndexValueRowID()
343 auto value_rowids_val = [value_rowids](int index) { in CalculateOutputIndexValueRowID()
344 switch (value_rowids.type) { in CalculateOutputIndexValueRowID()
346 return static_cast<int>(value_rowids.data.i32[index]); in CalculateOutputIndexValueRowID()
348 return static_cast<int>(value_rowids.data.i64[index]); in CalculateOutputIndexValueRowID()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RaggedTensorToTensor.pbtxt9 * "VALUE_ROWIDS": the value_rowids tensor from the ragged tensor.
10 * "FIRST_DIM_SIZE": if value_rowids is used for the first dimension, then it
77 * "VALUE_ROWIDS": the value_rowids tensor from the ragged tensor.
78 * "FIRST_DIM_SIZE": if value_rowids is used for the first dimension, then it
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor.pbtxt93 …argspec: "args=[\'cls\', \'values\', \'value_rowids\', \'nrows\', \'name\', \'validate\'], varargs…
144 name: "value_rowids"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor.pbtxt93 …argspec: "args=[\'cls\', \'values\', \'value_rowids\', \'nrows\', \'name\', \'validate\'], varargs…
144 name: "value_rowids"
/external/tensorflow/tensorflow/python/keras/engine/
Dragged_keras_tensor_test.py181 inp, value_rowids=[0, 0, 0, 0, 2, 2, 2, 3], nrows=5)
186 x, value_rowids=[0, 0, 0, 0, 2, 2, 2, 3], nrows=5)