Home
last modified time | relevance | path

Searched defs:row_partitions (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/structured/
H A Dstructured_tensor.py116 def _old_init(cls, fields, shape, nrows, row_partitions, internal=False): argument
162 row_partitions=None, argument
573 def row_partitions(self): member in StructuredTensor
1199 def _find_shape_dtype(fields, nrows, row_partitions): argument
1282 def _merge_row_partitions(row_partitions, value, rank, dtype, validate): argument
1368 def _empty_dict_pylist_from_row_partitions(row_partitions, nrows): argument
1669 def _dynamic_ragged_shape_init(fields, shape, nrows, row_partitions): argument
H A Dstructured_array_ops_test.py251 def testSizeObject(self, row_partitions, shape, dtype, expected): argument
417 def testZerosLikeObject(self, row_partitions, shape, dtype, expected): argument
567 def testOnesLikeObject(self, row_partitions, shape, dtype, expected): argument
667 def testRank(self, row_partitions, shape, expected): argument
H A Dstructured_tensor_test.py557 row_partitions=None): argument
689 row_partitions=None, argument
H A Dstructured_tensor_dynamic.py25 def _dynamic_ragged_shape_init(fields, shape, nrows, row_partitions): argument
H A Dstructured_array_ops.py485 def _structured_tensor_from_row_partitions(shape, row_partitions): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Ddynamic_ragged_shape_test.py217 def testFromTensor(self, value, row_partitions, inner_shape): argument
290 row_partitions, argument
1041 row_partitions, argument
2914 row_partitions, argument
3615 def test_constructor_idempotent(self, row_partitions, static_inner_shape, argument
3791 def test_constructor_raises(self, row_partitions, static_inner_shape, argument
H A Ddynamic_ragged_shape.py407 def from_row_partitions(cls, row_partitions, dtype=None): argument
440 def row_partitions(self): member in DynamicRaggedShape