Home
last modified time | relevance | path

Searched refs:tensor_handle_to_tensor_pointer (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dshim_common.cpp165 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); \
219 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_delete_tensor_object()
228 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_get_data_ptr()
241 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_get_storage_size()
248 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_get_dim()
257 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_get_numel()
266 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_get_storage_numel()
280 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_get_sizes()
291 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_get_size()
300 at::Tensor* t = tensor_handle_to_tensor_pointer(tensor); in aoti_torch_get_strides()
[all …]
H A Doss_proxy_executor.cpp8 at::Tensor* tensor_handle_to_tensor_pointer(AtenTensorHandle handle) { in tensor_handle_to_tensor_pointer() function
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()
413 at::Tensor* tensor = tensor_handle_to_tensor_pointer( 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 Dutils.h28 inline at::Tensor* tensor_handle_to_tensor_pointer(AtenTensorHandle handle) { in tensor_handle_to_tensor_pointer() function
77 return ptr ? std::make_optional(*tensor_handle_to_tensor_pointer(*ptr)) in pointer_to_optional()
84 return ptr ? std::make_optional(*tensor_handle_to_tensor_pointer(*ptr)) in pointer_to_optional()
151 result.emplace_back(*tensor_handle_to_tensor_pointer(*ptr));
H A Dtensor_converter.cpp34 at::Tensor tensor = *tensor_handle_to_tensor_pointer(handles[i]); in alloc_tensors_by_stealing_from_handles()
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runner/
H A Dpybind.cpp68 return *torch::aot_inductor::tensor_handle_to_tensor_pointer( in initAOTIRunnerBindings()