Searched defs:num_row_partitions (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | dynamic_ragged_shape_test.py | 292 num_row_partitions=None): argument
327 def testFromLengthsError(self, lengths, msg, num_row_partitions=None): argument
490 def testWithNumRowPartitionsRaises(self, lengths, num_row_partitions, argument
555 num_row_partitions=None): argument
597 num_row_partitions=None): argument
750 def testIsUniformTrue(self, lengths, axis, num_row_partitions=None): argument
764 def testIsUniformFalse(self, lengths, num_row_partitions, axis): argument
790 def testIsUniformRaises(self, lengths, num_row_partitions, axis, error_type, argument
835 def testGetItem(self, lengths, axis, expected, num_row_partitions=None): argument
2056 num_row_partitions=None): argument
[all …]
|
| H A D | dynamic_ragged_shape.py | 305 num_row_partitions=None, argument
445 def num_row_partitions(self): member in DynamicRaggedShape
819 def _with_num_row_partitions(self, num_row_partitions): argument
1359 def num_row_partitions(self) -> int: member in DynamicRaggedShape.Spec
2945 def _fix_start_index(index, rank, num_row_partitions): argument
|