Home
last modified time | relevance | path

Searched defs:keep_initializers_as_inputs (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport.cpp512 bool keep_initializers_as_inputs, in GraphEncoder()
741 bool keep_initializers_as_inputs, in EncodeGraph()
763 bool keep_initializers_as_inputs, in EncodeBlock()
1371 bool keep_initializers_as_inputs, in pretty_print_onnx()
1411 bool keep_initializers_as_inputs, in export_onnx()
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_custom_ops.py14 keep_initializers_as_inputs = False variable in TestCustomAutogradFunction
87 keep_initializers_as_inputs = False variable in TestExportAsContribOps
H A Dtest_models.py51 keep_initializers_as_inputs = False variable in TestModels
H A Dtest_autograd_funs.py15 keep_initializers_as_inputs = False variable in TestAutogradFuns
H A Dverify.py239 keep_initializers_as_inputs=True, argument
H A Donnx_test_common.py129 keep_initializers_as_inputs = True # For IR version 3 type export. variable in _TestONNXRuntime
/aosp_15_r20/external/pytorch/torch/onnx/
H A D_experimental.py25 keep_initializers_as_inputs: Optional[bool] = None variable in ExportOptions
H A Dutils.py1221 keep_initializers_as_inputs=None, argument
1474 keep_initializers_as_inputs=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()