Searched defs:rt_shape (Results 1 – 1 of 1) sorted by relevance
378 def testUniformSplitDynamicShape(self, rt_shape): argument398 def testRaggedSplitDynamicShape(self, rt_shape, lengths_shape): argument459 rt_shape, argument