Searched refs:wrap_pybind_function (Results 1 – 3 of 3) sorted by relevance
45 ::torch::wrap_pybind_function( in initONNXBindings()62 wrap_pybind_function(ONNXFunctionCallSubstitution)) in initONNXBindings()65 wrap_pybind_function(ONNXAutogradFunctionProcess)) in initONNXBindings()68 ::torch::wrap_pybind_function([](std::shared_ptr<Graph>& graph, in initONNXBindings()75 ::torch::wrap_pybind_function(PreprocessForONNX)) in initONNXBindings()78 ::torch::wrap_pybind_function( in initONNXBindings()87 ::torch::wrap_pybind_function(CastAllConstantToFloating)) in initONNXBindings()90 ::torch::wrap_pybind_function( in initONNXBindings()103 ::torch::wrap_pybind_function( in initONNXBindings()114 ::torch::wrap_pybind_function([](std::shared_ptr<Graph>& graph, in initONNXBindings()[all …]
374 auto wrap_pybind_function(Func&& f) { in wrap_pybind_function() function
1818 torch::wrap_pybind_function(at::init_num_threads), in initModule()