Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A Dio_adapter.py333 class RemoveNonTensorInputStep(InputAdaptStep): class
H A D_exporter_legacy.py1292 options.fx_tracer.input_adapter.append_step(io_adapter.RemoveNonTensorInputStep())