Searched refs:InputWithDtypeModel (Results 1 – 1 of 1) sorted by relevance
994 class InputWithDtypeModel(torch.jit.ScriptModule): class1001 InputWithDtypeModel(), x, input_names=["x"], dynamic_axes={"x": [0, 1]}1003 self.run_test(InputWithDtypeModel(), x, remained_onnx_input_idx=[])