Searched defs:apply_weights (Results 1 – 2 of 2) sorted by relevance
183 def apply_weights(self, state_dict: dict[str, torch.Tensor]) -> None: member in ONNXProgram
740 def apply_weights(self, state_dict: dict[str, torch.Tensor]) -> None: member in ONNXProgram