Home
last modified time | relevance | path

Searched defs:onnx_model (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/onnx/dynamo/
H A Dtest_dynamo_with_onnxruntime_backend.py766 def record_onnx_model_transform(onnx_model): argument
800 def record_onnx_model_transform(onnx_model): argument
826 def replace_relu_with_sigmoid(onnx_model): argument
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dcommon.py1822 onnx_model: Optional[OnnxModel] = None variable in OnnxContext
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py4128 def assertAtenOp(self, onnx_model, operator, overload_name=""): argument