Searched refs:MNISTModel (Results 1 – 2 of 2) sorted by relevance
105 class MNISTModel(nn.Module): class127 onnx_program = dynamo_export(MNISTModel(), tensor_x)332 class MNISTModel(nn.Module): class349 model = MNISTModel()695 class MNISTModel(nn.Module): class716 state_dict = MNISTModel().state_dict()719 model = MNISTModel()725 model = MNISTModel()
269 class MNISTModel(nn.Module): class292 MNISTModel(), (tensor_x,)