Home
last modified time | relevance | path

Searched defs:ragged_rank (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_factory_ops.py39 def constant(pylist, dtype=None, ragged_rank=None, inner_shape=None, argument
92 def constant_value(pylist, dtype=None, ragged_rank=None, inner_shape=None, argument
150 def _constant_value(ragged_factory, inner_factory, pylist, dtype, ragged_rank, argument
277 def _default_inner_shape_for_pylist(pylist, ragged_rank): argument
318 def placeholder(dtype, ragged_rank, value_shape=None, name=None): argument
Dragged_const_op_test.py181 ragged_rank=None, argument
299 ragged_rank=None, argument
388 ragged_rank, argument
Dragged_to_tensor_op_test.py327 ragged_rank=None, argument
395 ragged_rank=None, argument
621 ragged_rank=None): argument
753 ragged_rank=None, argument
829 def _generateRaggedTensor(self, shape, ragged_rank, dtype, fill=None, axis=0): argument
Dragged_constant_value_op_test.py185 ragged_rank=None, argument
304 ragged_rank=None, argument
Dragged_one_hot_op_test.py93 ragged_rank=None): argument
127 ragged_rank=None): argument
Dragged_from_tensor_op_test.py356 ragged_rank=1, argument
628 ragged_rank=1, argument
Dragged_size_op_test.py42 def testRaggedSize(self, test_input, size, ragged_rank=None): argument
Dragged_rank_op_test.py81 def testRaggedRank(self, test_input, expected_rank, ragged_rank=None): argument
Dragged_batch_gather_op_test.py353 expected_ragged_rank=None, ragged_rank=None, default_value='$NONE^'): argument
462 self, descr, params, indices, default_value, error, ragged_rank=None, argument
Dragged_eager_test.py37 def testRaggedTensorToList(self, pylist, ragged_rank=None): argument
Dragged_expand_dims_op_test.py114 ragged_rank=None, argument
Dragged_tile_op_test.py191 ragged_rank=None): argument
Dragged_reverse_op_test.py75 def testReverse(self, descr, data, axis, expected, ragged_rank=None): argument
Dragged_placeholder_op_test.py53 def testRaggedPlaceholder(self, dtype, ragged_rank, value_shape, name, argument
Dragged_row_lengths_op_test.py118 ragged_rank=None, argument
Dragged_merge_dims_op_test.py265 ragged_rank=None): argument
Dragged_tensor_value.py84 def ragged_rank(self): member in RaggedTensorValue
Dragged_print_op_test.py161 def testRaggedToString(self, rt, expected, summarize=None, ragged_rank=None): argument
Dragged_concat_ops.py300 def _increase_ragged_rank_to(rt_input, ragged_rank, row_splits_dtype): argument
Dragged_conversion_ops.py33 ragged_rank=1, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dunicode_decode_op_test.py133 def testBasicDecode(self, texts, ragged_rank=None): argument
149 def testBasicDecodeWithOffsets(self, texts, ragged_rank=None): argument
224 def testDecodeWithPaddedOutput(self, texts, expected, ragged_rank=None): argument
485 def testBasicSplit(self, texts, ragged_rank=None): argument
501 def testBasicSplitWithOffsets(self, texts, ragged_rank=None): argument
592 def testSplitWithPaddedOutput(self, texts, expected, ragged_rank=None): argument
/external/tensorflow/tensorflow/core/util/
Dragged_to_dense_util_test.cc34 int ragged_rank = 1; in TEST() local
48 int ragged_rank = 1; in TEST() local
65 int ragged_rank = 1; in TEST() local
Dragged_to_dense_util.cc41 int ragged_rank, const TensorShapeProto& shape, in CombineRaggedTensorToTensorShapes()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc47 int ragged_rank = 0; member
229 int ragged_rank, const RuntimeShape& output_shape, in CombineRaggedTensorToTensorShapes()
251 TfLiteNode* node, int first_dimension, int ragged_rank, in CalculateOutputSize()
450 void SetOutputT(TfLiteContext* context, int ragged_rank, in SetOutputT()
525 void SetOutput(TfLiteContext* context, int ragged_rank, in SetOutput()
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_from_variant_op.cc31 const Tensor& encoded_variant, int ragged_rank, DataType value_dtype, in RaggedComponentsFromVariant()
266 int ragged_rank = ragged_tensor.ragged_rank(); in ReturnRaggedTensor() local

12