Searched refs:test_vmap_chunksize (Results 1 – 1 of 1) sorted by relevance
3371 def test_vmap_chunksize(self, in_dim, out_dim, randomness): member in TestVmapOperators6035 def test_vmap_chunksize(self, in_dim, out_dim): member in TestRandomness