Searched refs:makeTensorWrapper (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | ADInterpreters.cpp | 43 return makeTensorWrapper(tensor, current_level, /*is_immutable=*/true); in materializeGradWrappers()
50 return makeTensorWrapper(tensor, current_level, /*is_immutable=*/true); in materializeGradWrappers()
119 return makeTensorWrapper(tensor, interpreter, is_immutable); in autogradBasedTransformSendToNext()
|
| H A D | TensorWrapper.h | 97 TORCH_API Tensor makeTensorWrapper(const Tensor& tensor, int64_t level, bool is_immutable=false);
98 TORCH_API Tensor makeTensorWrapper(const Tensor& tensor, const Interpreter& interpreter, bool is_im…
|
| H A D | TensorWrapper.cpp | 92 Tensor makeTensorWrapper(const Tensor& tensor, int64_t level, bool is_immutable) { in makeTensorWrapper() function
101 Tensor makeTensorWrapper(const Tensor& tensor, const Interpreter& interpreter, bool is_immutable) { in makeTensorWrapper() function
|
| /aosp_15_r20/external/pytorch/torch/csrc/functorch/ |
| H A D | init.cpp | 196 return makeTensorWrapper(self, level); in _wrap_for_grad()
|