Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dinliner.cpp15 GraphFunction* tryToGraphFunction(Node* n) { in tryToGraphFunction() function
20 return tryToGraphFunction(*fun_type->function()); in tryToGraphFunction()
26 return tryToGraphFunction(function); in tryToGraphFunction()
38 if (auto graphFunction = tryToGraphFunction(cur)) { in inlineCalls()
76 if (auto graphFunction = tryToGraphFunction(cur)) { in inlineCalls()
H A Dinliner.h10 TORCH_API GraphFunction* tryToGraphFunction(Node* n);
H A Dfreeze_module.cpp476 if (auto graphFunction = tryToGraphFunction(function)) { in inlineInterfaceCall()
/aosp_15_r20/external/pytorch/torch/csrc/jit/api/
H A Dfunction_impl.h176 TORCH_API GraphFunction* tryToGraphFunction(Function&) noexcept;
H A Dfunction_impl.cpp161 GraphFunction* tryToGraphFunction(Function& function) noexcept { in tryToGraphFunction() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfunction_substitution.cpp140 if (auto graphFunction = tryToGraphFunction(function)) { in functionCallSubstitution()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dhelper.cpp555 auto graphFunc = tryToGraphFunction(*func); in getCallFunctionGraph()
/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/
H A Dalias_analysis.cpp739 auto g = tryToGraphFunction(node); in analyzeImpl()