Home
last modified time | relevance | path

Searched refs:from_row_lengths (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_test.py216 lambda: [row_partition.RowPartition.from_row_lengths([3, 7])],
270 row_partition.RowPartition.from_row_lengths([3]),
271 row_partition.RowPartition.from_row_lengths([2, 0, 1]),
415 [row_partition.RowPartition.from_row_lengths([3, 4])],
518 row_partition.RowPartition.from_row_lengths([1, 0, 3, 0]))
699 row_partition.RowPartition.from_row_lengths([3, 2])]),
706 row_partition.RowPartition.from_row_lengths([2, 1]),
707 row_partition.RowPartition.from_row_lengths([2, 3, 1])]),
714 row_partition.RowPartition.from_row_lengths([2]),
715 row_partition.RowPartition.from_row_lengths([0, 0])]),
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition_test.py52 rt2 = RowPartition.from_row_lengths(row_lengths=[4, 0, 3, 1, 0])
239 rp = RowPartition.from_row_lengths(row_lengths, validate=False)
375 'factory': RowPartition.from_row_lengths,
380 'factory': RowPartition.from_row_lengths,
485 ('FromRowLengths', lambda: RowPartition.from_row_lengths([3, 0, 8]),
538 y=lambda: RowPartition.from_row_lengths([3, 5]),
612 x=lambda: RowPartition.from_row_lengths([2, 0, 2]),
613 y=lambda: RowPartition.from_row_lengths([2, 0, 2, 1]),
650 x=lambda: RowPartition.from_row_lengths([2, 0, 2]),
651 y=lambda: RowPartition.from_row_lengths([0, 0, 2]),
[all …]
Dragged_tensor_test.py78 rt2 = RaggedTensor.from_row_lengths(values, row_lengths=[4, 0, 3, 1, 0])
366 rt = RaggedTensor.from_row_lengths(values, row_lengths, validate=False)
383 rt = RaggedTensor.from_row_lengths([1, 2, 3, 4],
386 rt2 = RaggedTensor.from_row_lengths(rt, [2, 1, 0])
682 rt3 = RaggedTensor.from_row_lengths(values, row_lengths)
952 'factory': RaggedTensor.from_row_lengths,
1105 'factory': RaggedTensor.from_row_lengths,
1111 'factory': RaggedTensor.from_row_lengths,
1117 'factory': RaggedTensor.from_row_lengths,
1123 'factory': RaggedTensor.from_row_lengths,
Dragged_concat_ops.py167 return ragged_tensor.RaggedTensor.from_row_lengths(
Dragged_tensor.py454 def from_row_lengths(cls, values, row_lengths, name=None, validate=True): member in RaggedTensor
489 row_partition = RowPartition.from_row_lengths(
781 result = cls.from_row_lengths(result, lengths, validate=validate)
Dragged_tensor_supported_values_test.py178 rt = RaggedTensor.from_row_lengths(values, row_lengths)
Dragged_array_ops.py192 masked_values = ragged_tensor.RaggedTensor.from_row_lengths(
Dragged_tensor_shape.py552 rt_input = ragged_tensor.RaggedTensor.from_row_lengths(rt_input, [nrows],
Drow_partition.py356 def from_row_lengths(cls, row_lengths, validate=True, preferred_dtype=None): member in RowPartition
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Drebatch_dataset_test.py372 ragged_tensor.RaggedTensor.from_row_lengths(
382 ragged_tensor.RaggedTensor.from_row_lengths(list(range(3)), [1, 2]),
383 ragged_tensor.RaggedTensor.from_row_lengths(list(range(3, 10)),
543 ragged_tensor.RaggedTensor.from_row_lengths(values, row_lengths))
557 ragged_tensor.RaggedTensor.from_row_lengths(
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_cpu_test.py220 feature0 = ragged_tensor.RaggedTensor.from_row_lengths(
223 feature1 = ragged_tensor.RaggedTensor.from_row_lengths(
226 feature2 = ragged_tensor.RaggedTensor.from_row_lengths(
Dtpu_embedding_v2_test.py1077 ragged_tensor.RaggedTensor.from_row_lengths(
1080 ragged_tensor.RaggedTensor.from_row_lengths(
1083 ragged_tensor.RaggedTensor.from_row_lengths(
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py349 ragged_tensor = ragged_tensor_lib.RaggedTensor.from_row_lengths(
410 ragged_tensor = ragged_tensor_lib.RaggedTensor.from_row_lengths(
472 ragged_tensor = ragged_tensor_lib.RaggedTensor.from_row_lengths(
Dinput_lib_test.py1049 ragged_tensor = ragged_tensor_lib.RaggedTensor.from_row_lengths(
1170 ragged_tensor = ragged_tensor_lib.RaggedTensor.from_row_lengths(
1218 ragged_tensor = ragged_tensor_lib.RaggedTensor.from_row_lengths(
/external/tensorflow/tensorflow/python/ops/
Dparsing_config.py784 return ragged_tensor.RaggedTensor.from_row_lengths(
864 ragged_tensor.RaggedTensor.from_row_lengths(
871 return ragged_tensor.RaggedTensor.from_row_lengths(
/external/tensorflow/tensorflow/python/keras/engine/
Dragged_keras_tensor_test.py212 out = ragged_tensor.RaggedTensor.from_row_lengths(
217 expected = ragged_tensor.RaggedTensor.from_row_lengths(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor.pbtxt64 name: "from_row_lengths"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor.pbtxt64 name: "from_row_lengths"
/external/tensorflow/tensorflow/python/keras/layers/
Dwrappers.py252 y = nest.map_structure(ragged_tensor.RaggedTensor.from_row_lengths, y,
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmap_test.py827 sparse = ragged_tensor.RaggedTensor.from_row_lengths(
838 sparse = ragged_tensor.RaggedTensor.from_row_lengths(
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py1475 ('RaggedTensorRaggedRank1', ragged_tensor.RaggedTensor.from_row_lengths,
1508 ragged_tensor.RaggedTensor.from_row_lengths,
1517 ragged_tensor.RaggedTensor.from_row_lengths,
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py2191 ragged_tensor.RaggedTensor.from_row_lengths(