Searched refs:NNLossFunctionModule (Results 1 – 2 of 2) sorted by relevance
21 NNLossFunctionModule,85 "loss_function_ops": NNLossFunctionModule(),
331 class NNLossFunctionModule(torch.nn.Module): class