Home
last modified time | relevance | path

Searched refs:from_nested_row_splits (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_from_tensor_op_test.py368 self.assertAllEqual(rt, RaggedTensor.from_nested_row_splits(
384 self.assertAllEqual(rt, RaggedTensor.from_nested_row_splits(
501 self.assertAllEqual(rt, RaggedTensor.from_nested_row_splits(
561 self.assertAllEqual(rt, RaggedTensor.from_nested_row_splits(
Dragged_tensor_supported_values_test.py164 rt = RaggedTensor.from_nested_row_splits(values, nested_row_splits)
178 rt = RaggedTensor.from_nested_row_splits(values, nested_row_splits)
256 wrapped_x = ragged_tensor.RaggedTensor.from_nested_row_splits(
287 wrapped_x = ragged_tensor.RaggedTensor.from_nested_row_splits(
289 wrapped_y = ragged_tensor.RaggedTensor.from_nested_row_splits(
425 rt = RaggedTensor.from_nested_row_splits(wrapped_tensor, nested_row_splits)
Dragged_getitem_test.py390 rt = RaggedTensor.from_nested_row_splits(
416 rt = RaggedTensor.from_nested_row_splits(
479 rt = RaggedTensor.from_nested_row_splits(values, [splits1, splits2])
Dragged_squeeze_op.py113 squeezed_rt = RaggedTensor.from_nested_row_splits(squeezed_rt,
Dragged_math_ops.py331 ones = ragged_tensor.RaggedTensor.from_nested_row_splits(
351 ones = ragged_tensor.RaggedTensor.from_nested_row_splits(
679 ones = ragged_tensor.RaggedTensor.from_nested_row_splits(
687 return ragged_tensor.RaggedTensor.from_nested_row_splits(
Dragged_tensor_test.py100 rt = RaggedTensor.from_nested_row_splits(
563 rt = RaggedTensor.from_nested_row_splits(
587 rt = RaggedTensor.from_nested_row_splits(
610 RaggedTensor.from_nested_row_splits(
747 rt1 = RaggedTensor.from_nested_row_splits(values, nested_row_splits)
891 rt2 = RaggedTensor.from_nested_row_splits(
1789 rt1 = RaggedTensor.from_nested_row_splits(
Dragged_concat_ops.py240 return ragged_tensor.RaggedTensor.from_nested_row_splits(
Dragged_array_ops.py148 masked_values = ragged_tensor.RaggedTensor.from_nested_row_splits(
252 return ragged_tensor.RaggedTensor.from_nested_row_splits(
Dragged_gather_ops.py154 result = ragged_tensor.RaggedTensor.from_nested_row_splits(
Dragged_tensor.py724 def from_nested_row_splits(cls, member in RaggedTensor
2019 return cls.from_nested_row_splits(
2693 return RaggedTensor.from_nested_row_splits(
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RaggedGather.pbtxt69 ragged.from_nested_row_splits(params_dense_values, params_nested_splits)`
74 ragged.from_nested_row_splits(output_dense_values, output_nested_splits)`
Dapi_def_RaggedTensorToSparse.pbtxt38 input=ragged.from_nested_row_splits(rt_dense_values, rt_nested_splits)
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dtflite_text_api.py114 ragged_tensor = tf.RaggedTensor.from_nested_row_splits(
Dsentencepiece_tokenizer.py77 tokens = tf.RaggedTensor.from_nested_row_splits(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor.pbtxt56 name: "from_nested_row_splits"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor.pbtxt56 name: "from_nested_row_splits"
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dngrams_test.py75 input_tensor = tf.RaggedTensor.from_nested_row_splits(
/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
Dunicode_encode_op_test.py276 test_value = ragged_tensor.RaggedTensor.from_nested_row_splits(
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py1522 dummy_tensor = ragged_tensor.RaggedTensor.from_nested_row_splits(
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td11422 ragged.from_nested_row_splits(params_dense_values, params_nested_splits)`
11427 ragged.from_nested_row_splits(output_dense_values, output_nested_splits)`