Searched refs:test_chunk_vmap (Results 1 – 1 of 1) sorted by relevance
3345 def test_chunk_vmap(self, in_dim, out_dim, randomness): member in TestVmapOperators6014 def test_chunk_vmap(self, in_dim, out_dim): member in TestRandomness