Searched refs:replace_tile_positional_embedding (Results 1 – 4 of 4) sorted by relevance
14 replace_tile_positional_embedding,67 self.model = replace_tile_positional_embedding(self.model)
14 replace_tile_positional_embedding,123 m = replace_tile_positional_embedding(m)
8 replace_tile_positional_embedding,
654 def replace_tile_positional_embedding(model: nn.Module) -> nn.Module: function686 replace_tile_positional_embedding(module)