Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dgraph_opt.h103 TORCH_API bool isGraphCompilable(const std::shared_ptr<Graph>& graph);
H A Dgraph_opt.cpp262 bool isGraphCompilable(const std::shared_ptr<Graph>& graph) { in isGraphCompilable() function
H A Dtensorexpr_init.cpp951 te.def("is_graph_compilable", &tensorexpr::isGraphCompilable); in initTensorExprBindings()