Searched refs:input_module_path (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/ |
| H A D | TracerRunner.cpp | 227 const std::string& input_module_path, in run_model() argument
235 torch::jit::mobile::MobileModelRunner module_runner(input_module_path, 0); in run_model()
255 input_module_path); in run_model()
305 TracerResult trace_run(const std::string& input_module_path) { in trace_run() argument
306 return trace_run(std::vector<std::string>(1, input_module_path)); in trace_run()
326 for (auto& input_module_path : input_module_paths) { in trace_run() local
331 input_module_path, root_ops, enabled_backends, called_kernel_tags); in trace_run()
338 input_module_path, root_ops, enabled_backends, called_kernel_tags); in trace_run()
348 input_module_path, root_ops, enabled_backends, called_kernel_tags); in trace_run()
|
| H A D | TracerRunner.h | 32 TracerResult trace_run(const std::string& input_module_path);
|