Home
last modified time | relevance | path

Searched defs:custom_opsets (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport.cpp513 const std::map<std::string, int>& custom_opsets, in GraphEncoder()
1372 const std::map<std::string, int>& custom_opsets, in pretty_print_onnx()
1412 const std::map<std::string, int>& custom_opsets, in export_onnx()
/aosp_15_r20/external/pytorch/torch/onnx/
H A D_experimental.py26 custom_opsets: Optional[Mapping[str, int]] = None variable in ExportOptions
H A Dutils.py1222 custom_opsets=None, argument
1476 custom_opsets=None, argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpython_ir.cpp260 const NodeAttrNameMap& node_attr_to_name) { in initPythonIRBindings()
322 bool add_node_names) { in initPythonIRBindings()