Searched refs:ArithmeticModule (Results 1 – 2 of 2) sorted by relevance
326 class ArithmeticModule(torch.nn.Module): class338 arithmetic_module = ArithmeticModule()
1739 class ArithmeticModule(torch.nn.Module): class1748 self.run_test(ArithmeticModule(), x)1751 class ArithmeticModule(torch.nn.Module): class1761 self.run_test(ArithmeticModule(), (x, y))1763 class ArithmeticModule(torch.nn.Module): class1770 self.run_test(ArithmeticModule(), x, remained_onnx_input_idx=[])1774 class ArithmeticModule(torch.nn.Module): class1784 self.run_test(ArithmeticModule(), (x, y))1786 class ArithmeticModule(torch.nn.Module): class1793 self.run_test(ArithmeticModule(), x, remained_onnx_input_idx=[])[all …]