Home
last modified time | relevance | path

Searched refs:from_row_splits (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition_test.py46 rp = RowPartition.from_row_splits(row_splits=[0, 4, 4, 7, 8, 8])
51 rt1 = RowPartition.from_row_splits(row_splits=[0, 4, 4, 7, 8, 8])
62 inner_rt = RowPartition.from_row_splits(row_splits=[0, 4, 4, 7, 8, 8])
63 outer_rt = RowPartition.from_row_splits(row_splits=[0, 3, 3, 5])
176 rp = RowPartition.from_row_splits(row_splits, validate=False)
191 rt1 = RowPartition.from_row_splits(splits1)
192 rt2 = RowPartition.from_row_splits(splits2)
193 rt3 = RowPartition.from_row_splits(splits3)
194 rt4 = RowPartition.from_row_splits(splits4)
195 rt5 = RowPartition.from_row_splits(splits5)
[all …]
Dragged_to_sparse_op_test.py146 bad_rt1 = ragged_tensor.RaggedTensor.from_row_splits(
152 bad_rt2 = ragged_tensor.RaggedTensor.from_row_splits(
154 bad_rt3 = ragged_tensor.RaggedTensor.from_row_splits(
156 values=ragged_tensor.RaggedTensor.from_row_splits(
165 bad_rt4 = ragged_tensor.RaggedTensor.from_row_splits(
167 values=ragged_tensor.RaggedTensor.from_row_splits(
175 bad_rt5 = ragged_tensor.RaggedTensor.from_row_splits(
Dragged_tensor_test.py70 rt = RaggedTensor.from_row_splits(
77 rt1 = RaggedTensor.from_row_splits(values, row_splits=[0, 4, 4, 7, 8, 8])
88 inner_rt = RaggedTensor.from_row_splits(
90 outer_rt = RaggedTensor.from_row_splits(
105 rt = RaggedTensor.from_row_splits(
152 rp = RowPartition.from_row_splits(row_splits)
161 rp = RowPartition.from_row_splits(row_splits)
286 rt = RaggedTensor.from_row_splits(values, row_splits, validate=False)
308 rt1 = RaggedTensor.from_row_splits(values, splits1)
309 rt2 = RaggedTensor.from_row_splits(values, splits2)
[all …]
Dragged_getitem_test.py252 rt = RaggedTensor.from_row_splits(EXAMPLE_RAGGED_TENSOR_2D_VALUES,
314 rt = RaggedTensor.from_row_splits(EXAMPLE_RAGGED_TENSOR_2D_VALUES,
433 rt = RaggedTensor.from_row_splits([], [0])
444 rt = RaggedTensor.from_row_splits([], [0])
463 rt = RaggedTensor.from_row_splits(EXAMPLE_RAGGED_TENSOR_2D_VALUES, splits)
475 rt = RaggedTensor.from_row_splits(values, [0, 1])
560 RaggedTensor.from_row_splits(EXAMPLE_RAGGED_TENSOR_3D_VALUES,
582 RaggedTensor.from_row_splits(EXAMPLE_RAGGED_TENSOR_3D_VALUES,
Dragged_concat_ops.py202 return ragged_tensor.RaggedTensor.from_row_splits(
282 return ragged_tensor.RaggedTensor.from_row_splits(
289 return ragged_tensor.RaggedTensor.from_row_splits(
Dragged_map_flat_values_op_test.py194 z = ragged_tensor.RaggedTensor.from_row_splits([0, 1, 2], z_splits)
214 x = ragged_tensor.RaggedTensor.from_row_splits([3, 1, 4, 1, 5], splits1)
215 y = ragged_tensor.RaggedTensor.from_row_splits([1, 2, 3, 4, 5], splits2)
Dragged_factory_ops.py82 return ragged_tensor.RaggedTensor.from_row_splits(values, row_splits,
349 result = ragged_tensor.RaggedTensor.from_row_splits(result, row_splits,
Dragged_string_ops.py179 ragged_input_tensor = ragged_tensor.RaggedTensor.from_row_splits(
453 codepoints = ragged_tensor.RaggedTensor.from_row_splits(
458 offsets = ragged_tensor.RaggedTensor.from_row_splits(
820 output = ragged_tensor.RaggedTensor.from_row_splits(
Dragged_array_ops.py167 return ragged_tensor.RaggedTensor.from_row_splits(
202 masked_values = ragged_tensor.RaggedTensor.from_row_splits(
839 return ragged_tensor.RaggedTensor.from_row_splits(
Dragged_math_ops.py111 return ragged_tensor.RaggedTensor.from_row_splits(
260 return ragged_tensor.RaggedTensor.from_row_splits(
Dragged_tensor.py407 def from_row_splits(cls, values, row_splits, name=None, validate=True): member in RaggedTensor
446 row_partition = RowPartition.from_row_splits(
741 result = cls.from_row_splits(result, splits, validate=validate)
1680 return cls.from_row_splits(values, splits, validate=False)
2387 RowPartition.from_row_splits(row_splits, validate=False),
2779 return RaggedTensor.from_row_splits(new_values, new_splits)
Dragged_segment_op_test.py211 segment_ids2 = ragged_tensor.RaggedTensor.from_row_splits(
Dragged_batch_gather_op_test.py476 ragged_indices = ragged_tensor.RaggedTensor.from_row_splits(
Dragged_getitem.py219 return ragged_tensor.RaggedTensor.from_row_splits(
/external/tensorflow/tensorflow/python/keras/engine/
Dinput_layer_test.py198 rt = ragged_tensor.RaggedTensor.from_row_splits(
206 rt = ragged_tensor.RaggedTensor.from_row_splits(
220 rt = ragged_tensor.RaggedTensor.from_row_splits(
229 rt = ragged_tensor.RaggedTensor.from_row_splits(
306 rt = ragged_tensor.RaggedTensor.from_row_splits(
314 rt = ragged_tensor.RaggedTensor.from_row_splits(
334 rt = ragged_tensor.RaggedTensor.from_row_splits(
343 rt = ragged_tensor.RaggedTensor.from_row_splits(
Dtraining_utils_v1_test.py411 ragged_tensor.RaggedTensor.from_row_splits(
445 tensor_1 = ragged_tensor.RaggedTensor.from_row_splits(
447 tensor_2 = ragged_tensor.RaggedTensor.from_row_splits(
Dragged_keras_tensor_test.py196 out = ragged_tensor.RaggedTensor.from_row_splits(
201 expected = ragged_tensor.RaggedTensor.from_row_splits(
/external/tensorflow/tensorflow/python/ops/
Dparsing_config.py732 rt = ragged_tensor.RaggedTensor.from_row_splits(
781 return ragged_tensor.RaggedTensor.from_row_splits(
822 return ragged_tensor.RaggedTensor.from_row_splits(
830 return ragged_tensor.RaggedTensor.from_row_splits(
887 ragged_tensor.RaggedTensor.from_row_splits(val, split, validate=False)
894 ragged_tensor.RaggedTensor.from_row_splits(val, split, validate=False)
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dragged_tensor_to_tensor_test.py27 ragged_tensor = tf.RaggedTensor.from_row_splits(
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RaggedRange.pbtxt36 result = tf.ragged.from_row_splits(rt_dense_values, rt_nested_splits)
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_test.py514 row_partition.RowPartition.from_row_splits([0, 2, 2, 3, 4]))
544 row_partition.RowPartition.from_row_splits([0, 1, 2]))
546 row_partition.RowPartition.from_row_splits([0, 1, 2]))
551 partition = row_partition.RowPartition.from_row_splits([0])
968 row_partition.RowPartition.from_row_splits([0, 1, 2]))
970 row_partition.RowPartition.from_row_splits([0, 1, 2]))
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils_test.py246 x = ragged_tensor.RaggedTensor.from_row_splits(dt, row_splits=[0, 1])
Dtf_utils_test.py198 tensor = ragged_tensor.RaggedTensor.from_row_splits(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor.pbtxt72 name: "from_row_splits"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor.pbtxt72 name: "from_row_splits"

12