Searched refs:proxy_executor (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| D | model.h | 140 AOTIProxyExecutorHandle proxy_executor) { in run() argument
149 model->run_impl(input_handles, output_handles, stream, proxy_executor); in run()
154 model->run_impl(input_handles, output_handles, stream, proxy_executor); in run()
161 AOTIProxyExecutorHandle proxy_executor,
175 model->const_run_impl(stream, proxy_executor, initialization);
580 AOTIProxyExecutorHandle proxy_executor,
586 AOTIProxyExecutorHandle proxy_executor);
597 AOTIProxyExecutorHandle proxy_executor);
603 AOTIProxyExecutorHandle proxy_executor);
|
| D | model_container.h | 80 AOTIProxyExecutorHandle proxy_executor) { in run() argument
93 stream, proxy_executor, /* initialization = */ true); in run()
105 model->run(input_handles, output_handles, stream, proxy_executor); in run()
161 AOTIProxyExecutorHandle proxy_executor) { in run_const_fold() argument
171 auto folded_const_map = model->run_const_fold(stream, proxy_executor); in run_const_fold()
194 auto folded_const_map = model->run_const_fold(stream, proxy_executor); in run_const_fold()
|
| /external/pytorch/torch/csrc/inductor/aoti_torch/ |
| D | shim_common.cpp | 1132 AOTIProxyExecutorHandle proxy_executor, in aoti_torch_proxy_executor_call_function() argument
1139 ProxyExecutor* executor = reinterpret_cast<ProxyExecutor*>(proxy_executor); in aoti_torch_proxy_executor_call_function()
|
| /external/pytorch/torch/csrc/inductor/aoti_torch/c/ |
| D | shim.h | 619 AOTIProxyExecutorHandle proxy_executor,
|