Home
last modified time | relevance | path

Searched refs:row_lengths (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_split_op_test.py43 row_lengths=[1, 3], # shape=(2, None)
51 row_lengths=[1, 2, 1], # shape=(3, None)
60 row_lengths=[1, 1, 1, 1, 1], # shape=(5, None)
71 row_lengths=[4, 0], # shape=(2, None)
79 row_lengths=[0, 0], # shape=(2, None)
87 row_lengths=[1, 3], # shape=(2, None)
94 row_lengths=[1, 3], # shape=(2, None)
102 row_lengths=[1, 3], # shape=(2, None, 2)
110 row_lengths=[1, 3], # shape=(2, None, 2)
119 row_lengths=[1, 3], # shape=(2, None)
[all …]
Drow_partition.py116 row_lengths=None, argument
174 for tensor in [row_lengths, value_rowids, nrows, uniform_row_length, nvals]:
181 self._row_lengths = row_lengths
294 row_lengths = bincount_ops.bincount(
299 row_splits = array_ops.concat([[0], math_ops.cumsum(row_lengths)], axis=0)
301 row_lengths.set_shape([const_nrows])
306 row_lengths=row_lengths,
376 row_lengths,
409 with ops.name_scope(None, "RowPartitionFromRowLengths", [row_lengths]):
410 row_lengths = cls._convert_row_partition(
[all …]
Dsegment_id_ops.py66 row_lengths = splits[1:] - splits[:-1]
69 return ragged_util.repeat(indices, repeats=row_lengths, axis=0)
120 row_lengths = bincount_ops.bincount(
125 splits = array_ops.concat([[0], math_ops.cumsum(row_lengths)], axis=0)
Dragged_batch_gather_with_default_op.py104 row_lengths = ragged_array_ops.expand_dims(
105 params.row_lengths(axis=num_batch_dimensions),
107 upper_bounds = math_ops.cast(row_lengths, indices.dtype)
Dragged_row_lengths_op_test.py117 lengths = rt.row_lengths(axis)
138 rt.row_lengths(axis)
Drow_partition_test.py60 rt2 = RowPartition.from_row_lengths(row_lengths=[4, 0, 3, 1, 0])
107 row_lengths=[2, 3, 4],
251 row_lengths = constant_op.constant([2, 0, 3, 1, 1], dtypes.int64)
253 rp = RowPartition.from_row_lengths(row_lengths, validate=False)
256 rp_row_lengths = rp.row_lengths()
259 self.assertIs(rp_row_lengths, row_lengths) # nrows
261 self.assertAllEqual(rp_row_lengths, row_lengths)
605 self.assertAllEqual(r.row_lengths(), result.row_lengths())
1202 test_class.assertAllEqual(actual.row_lengths(), expected.row_lengths())
Dragged_tensor_test.py80 rt2 = RaggedTensor.from_row_lengths(values, row_lengths=[4, 0, 3, 1, 0])
365 row_lengths = constant_op.constant([2, 0, 3, 1, 1], dtypes.int64)
367 rt = RaggedTensor.from_row_lengths(values, row_lengths, validate=False)
373 rt_row_lengths = rt.row_lengths()
377 self.assertIs(rt_row_lengths, row_lengths) # cached_nrows
379 self.assertAllEqual(rt_row_lengths, row_lengths)
388 self.assertAllEqual([2, 1, 0], rt2.row_lengths())
698 self.assertAllEqual(rt.row_lengths(), [2, 0, 3, 1, 1])
708 row_lengths = constant_op.constant([2, 0, 3, 1, 1])
711 rt3 = RaggedTensor.from_row_lengths(values, row_lengths)
[all …]
Dragged_tensor_supported_values_test.py149 row_lengths = constant_op.constant([2, 0, 3, 1, 2], dtypes.int64)
150 rt = RaggedTensor.from_row_lengths(values, row_lengths)
153 self.assertAllEqual(rt.row_lengths(), row_lengths)
Dragged_matmul_op_test.py113 values=T([10, 5, 7]), row_lengths=[3, 2, 0, 5]),
115 values=T([10, 7, 9]), row_lengths=[3, 2, 0, 5])),
Dragged_tensor.py461 def from_row_lengths(cls, values, row_lengths, name=None, validate=True): argument
496 with ops.name_scope(name, "RaggedFromRowLengths", [values, row_lengths]):
498 row_lengths=row_lengths,
1248 def row_lengths(self, axis=1, name=None): member in RaggedTensor
1279 return self._row_partition.row_lengths()
1290 return self.with_values(self.values.row_lengths(axis - 1))
1314 rt_nested_row_lengths.append(rt.row_lengths())
1356 result = math_ops.maximum(math_ops.reduce_max(self.row_lengths()), 0)
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py418 row_lengths = np.mod(np.arange(20), 4).astype(np.int64)
420 np.repeat(np.arange(20, dtype=np.float32), row_lengths), row_lengths)
592 row_lengths = np.mod(np.arange(20), 4).astype(np.int64)
594 np.repeat(np.arange(20, dtype=np.float32), row_lengths), row_lengths)
653 row_lengths = np.mod(np.arange(20), 4).astype(np.int64)
655 np.repeat(np.arange(20, dtype=np.float32), row_lengths), row_lengths)
714 row_lengths = np.mod(np.arange(50), 4).astype(np.int64)
716 np.repeat(np.arange(50, dtype=np.float32), row_lengths), row_lengths)
Dinput_lib_test.py1112 row_lengths = np.mod(np.arange(20), 4).astype(np.int64)
1114 np.repeat(np.arange(20, dtype=np.float32), row_lengths), row_lengths)
1232 row_lengths = np.mod(np.arange(20), 4).astype(np.int64)
1234 np.repeat(np.arange(20, dtype=np.float32), row_lengths), row_lengths)
1275 row_lengths = np.mod(np.arange(20), 4).astype(np.int64)
1277 np.repeat(np.arange(20, dtype=np.float32), row_lengths), row_lengths)
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_v2.py421 inputs, row_lengths = backend.convert_inputs_if_ragged(inputs)
422 is_ragged_input = (row_lengths is not None)
451 input_length=row_lengths if row_lengths is not None else timesteps,
458 inputs, initial_state, training, mask, row_lengths)
466 is_ragged_input, outputs, row_lengths, go_backwards=self.go_backwards)
1148 inputs, row_lengths = backend.convert_inputs_if_ragged(inputs)
1149 is_ragged_input = (row_lengths is not None)
1179 input_length=row_lengths if row_lengths is not None else timesteps,
1214 row_lengths,
1241 row_lengths
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.experimental.-row-partition.pbtxt24 …argspec: "args=[\'self\', \'row_splits\', \'row_lengths\', \'value_rowids\', \'nrows\', \'uniform_…
28 …argspec: "args=[\'cls\', \'row_lengths\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None, …
67 name: "row_lengths"
Dtensorflow.-ragged-tensor.pbtxt65 …argspec: "args=[\'cls\', \'values\', \'row_lengths\', \'name\', \'validate\'], varargs=None, keywo…
120 name: "row_lengths"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.-row-partition.pbtxt24 …argspec: "args=[\'self\', \'row_splits\', \'row_lengths\', \'value_rowids\', \'nrows\', \'uniform_…
28 …argspec: "args=[\'cls\', \'row_lengths\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None, …
67 name: "row_lengths"
Dtensorflow.-ragged-tensor.pbtxt65 …argspec: "args=[\'cls\', \'values\', \'row_lengths\', \'name\', \'validate\'], varargs=None, keywo…
120 name: "row_lengths"
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dngrams_test.cc122 std::vector<int64_t> row_lengths; in ExtractRowLengthsTensorVector() local
123 row_lengths.reserve(row_splits.size() - 1); in ExtractRowLengthsTensorVector()
127 row_lengths.push_back(tail - head); in ExtractRowLengthsTensorVector()
130 return row_lengths; in ExtractRowLengthsTensorVector()
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_for_serving_test.py160 reduced[-1] /= np.expand_dims(ragged.row_lengths().numpy(), axis=1)
197 row_lengths=self.feature_watched_row_lengths)
200 row_lengths=self.feature_favorited_row_lengths)
203 row_lengths=self.feature_friends_row_lengths)
/external/libtextclassifier/native/utils/
Dbert_tokenizer_test.cc85 EXPECT_THAT(results.row_lengths, ElementsAre(1, 1, 1, 5)); in TEST()
101 EXPECT_THAT(results.row_lengths, ElementsAre(1)); in TEST()
Dbert_tokenizer.h54 std::vector<int> row_lengths; member
/external/tensorflow/tensorflow/python/tpu/tests/
Dtpu_embedding_base_test.py270 row_lengths=self.feature_watched_row_lengths_high_dimensional,
273 row_lengths=self.feature_favorited_row_lengths_high_dimensional,
276 row_lengths=self.feature_friends_row_lengths_high_dimensional,
285 row_lengths=ragged.row_lengths(), values=values))
/external/tflite-support/tensorflow_lite_support/cc/text/tokenizers/
Dbert_tokenizer.cc87 result.row_lengths.emplace_back(num_word_pieces); in TokenizeWordpiece()
Dbert_tokenizer.h56 std::vector<int> row_lengths; member
/external/tensorflow/tensorflow/python/ops/
Dparsing_config.py858 rt.row_starts(), partition_t.row_lengths())
864 rt.row_starts(), partition_t.row_lengths())
876 math_ops.cumsum(nrows, exclusive=True), partition_t.row_lengths())

12