Home
last modified time | relevance | path

Searched refs:ArithmeticModule (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/executorch/backends/vulkan/test/
H A Dtest_vulkan_delegate.py326 class ArithmeticModule(torch.nn.Module): class
338 arithmetic_module = ArithmeticModule()
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py1739 class ArithmeticModule(torch.nn.Module): class
1748 self.run_test(ArithmeticModule(), x)
1751 class ArithmeticModule(torch.nn.Module): class
1761 self.run_test(ArithmeticModule(), (x, y))
1763 class ArithmeticModule(torch.nn.Module): class
1770 self.run_test(ArithmeticModule(), x, remained_onnx_input_idx=[])
1774 class ArithmeticModule(torch.nn.Module): class
1784 self.run_test(ArithmeticModule(), (x, y))
1786 class ArithmeticModule(torch.nn.Module): class
1793 self.run_test(ArithmeticModule(), x, remained_onnx_input_idx=[])
[all …]