Searched defs:pt_inputs (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | common.py | 924 def create_onnx_input_binded_fn(onnx_model: OnnxModel, pt_inputs, example_outputs): argument
935 def create_onnx_fn(onnx_model: OnnxModel, pt_inputs): argument
1346 def adapt_pt_inputs_to_onnx(self, pt_inputs) -> Mapping[str, np.ndarray]: argument
1394 def create_iobinding(self, pt_inputs, example_outputs): argument
1445 def run_and_serialize_inputs_outputs(self, pt_inputs): argument
1460 def run(self, pt_inputs): argument
1530 def format_pt_inputs(self, pt_inputs): argument
1602 def format_pt_inputs(self, pt_inputs): argument
|