Home
last modified time | relevance | path

Searched refs:flatten_tensor_args (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Doss_proxy_executor.cpp362 AtenTensorHandle* flatten_tensor_args) { in call_function() argument
385 tensor_handle_to_tensor_pointer(flatten_tensor_args[tensor_id++]); in call_function()
398 tensor_handle_to_tensor_pointer(flatten_tensor_args[tensor_id++]); in call_function()
414 flatten_tensor_args[tensor_id++]); in call_function()
465 tensor_handle_to_tensor_pointer(flatten_tensor_args[tensor_id++]); in call_function()
473 tensor_handle_to_tensor_pointer(flatten_tensor_args[tensor_id++]); in call_function()
H A Dproxy_executor.h19 AtenTensorHandle* flatten_tensor_args) = 0;
H A Doss_proxy_executor.h80 AtenTensorHandle* flatten_tensor_args) override;
H A Dshim_common.cpp1137 AtenTensorHandle* flatten_tensor_args) { in aoti_torch_proxy_executor_call_function() argument
1145 flatten_tensor_args); in aoti_torch_proxy_executor_call_function()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DVariableTypeManual.cpp136 set_history(flatten_tensor_args(result), grad_fn); in _fw_primal()
181 set_history(flatten_tensor_args(result), grad_fn); in _make_dual()
H A DVariableTypeUtils.h178 inline variable_list flatten_tensor_args(Args&&... args) { in flatten_tensor_args() function
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/c/
H A Dshim.h624 AtenTensorHandle* flatten_tensor_args);