Searched refs:TensorInputModel (Results 1 – 1 of 1) sorted by relevance
975 class TensorInputModel(torch.jit.ScriptModule): class982 TensorInputModel(), x, input_names=["x"], dynamic_axes={"x": [0, 1]}984 self.run_test(TensorInputModel(), x, remained_onnx_input_idx=[])