Searched refs:flatten_tensor_args (Results 1 – 7 of 7) sorted by relevance
362 AtenTensorHandle* flatten_tensor_args) { in call_function() argument385 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()
19 AtenTensorHandle* flatten_tensor_args) = 0;
80 AtenTensorHandle* flatten_tensor_args) override;
1137 AtenTensorHandle* flatten_tensor_args) { in aoti_torch_proxy_executor_call_function() argument1145 flatten_tensor_args); in aoti_torch_proxy_executor_call_function()
136 set_history(flatten_tensor_args(result), grad_fn); in _fw_primal()181 set_history(flatten_tensor_args(result), grad_fn); in _make_dual()
178 inline variable_list flatten_tensor_args(Args&&... args) { in flatten_tensor_args() function
624 AtenTensorHandle* flatten_tensor_args);