Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_numpy.cpp13 at::Tensor tensor_from_numpy( in tensor_from_numpy() function
215 at::Tensor tensor_from_numpy( in tensor_from_numpy() function
H A Dtensor_numpy.h9 at::Tensor tensor_from_numpy(PyObject* obj, bool warn_if_not_writeable = true);
H A Dtensor_new.cpp358 tensor_from_numpy(data, /*warn_if_not_writeable=*/!copy_numpy); in internal_new_from_data()
1723 tensor = tensor_from_numpy(arr, /*warn_if_not_writeable=*/false); in asarray()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_torch_functions_manual.cpp146 return THPVariable_Wrap(torch::utils::tensor_from_numpy(arg)); in THPVariable_from_numpy()
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp1117 auto other_tensor = torch::utils::tensor_from_numpy(other); in THPVariable___eq__()