Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_helper.py2158 cast_pytorch_to_onnx = { variable
2241 cast_pytorch_to_onnx["Byte"], # 0
2242 cast_pytorch_to_onnx["Char"], # 1
2243 cast_pytorch_to_onnx["Short"], # 2
2244 cast_pytorch_to_onnx["Int"], # 3
2245 cast_pytorch_to_onnx["Long"], # 4
2246 cast_pytorch_to_onnx["Half"], # 5
2247 cast_pytorch_to_onnx["Float"], # 6
2248 cast_pytorch_to_onnx["Double"], # 7
2249 cast_pytorch_to_onnx["Undefined"], # 8
[all …]