Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFunctionalizeFallbackKernel.cpp109 auto t_new = c10::IValue(at::functionalization::impl::to_functional_tensor(t)); in functionalizeFallback()
113 auto t_new = c10::IValue(at::functionalization::impl::to_functional_tensor(tensors)); in functionalizeFallback()
117 auto t_new = c10::IValue(at::functionalization::impl::to_functional_tensor(opt_tensors)); in functionalizeFallback()
193 return at::functionalization::impl::to_functional_tensor(out); in lift_functionalize()
204 return at::functionalization::impl::to_functional_tensor(out); in lift_fresh_functionalize()
225 return at::functionalization::impl::to_functional_tensor(out); in lift_fresh_functionalize_copy()
267 return at::functionalization::impl::to_functional_tensor(out); in _to_copy_functionalize()
H A DFunctionalTensorWrapper.cpp524 Tensor to_functional_tensor(const Tensor& tensor) { in to_functional_tensor() function
532 std::optional<Tensor> to_functional_tensor(const std::optional<Tensor>& tensor) { in to_functional_tensor() function
534 return std::make_optional<Tensor>(to_functional_tensor(*tensor)); in to_functional_tensor()
538 c10::List<::std::optional<Tensor>> to_functional_tensor(const c10::List<::std::optional<Tensor>>& t… in to_functional_tensor() function
542 outputs.push_back(to_functional_tensor(t_list[i])); in to_functional_tensor()
546 std::vector<Tensor> to_functional_tensor(ITensorListRef t_list) { in to_functional_tensor() function
550 outputs.push_back(to_functional_tensor(tensor)); in to_functional_tensor()
839 auto t_new = c10::IValue(at::functionalization::impl::to_functional_tensor(t)); in functionalize_op_helper()
846 auto t_new = c10::IValue(at::functionalization::impl::to_functional_tensor(tensors)); in functionalize_op_helper()
852 auto t_new = c10::IValue(at::functionalization::impl::to_functional_tensor(opt_tensors)); in functionalize_op_helper()
H A DFunctionalTensorWrapper.h307 TORCH_API Tensor to_functional_tensor(const Tensor& tensor);
308 TORCH_API std::optional<Tensor> to_functional_tensor(
310 TORCH_API c10::List<std::optional<Tensor>> to_functional_tensor(
312 TORCH_API std::vector<Tensor> to_functional_tensor(ITensorListRef t_list);
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_native_functions.cpp297 auto wrapped = at::functionalization::impl::to_functional_tensor(tensor); in empty_symint()
390 return at::functionalization::impl::to_functional_tensor(tensor); in lift()
395 return at::functionalization::impl::to_functional_tensor(tensor); in lift_fresh()
476 auto self_wrapped = at::functionalization::impl::to_functional_tensor(self); in logsumexp_out()
477 auto out_wrapped = at::functionalization::impl::to_functional_tensor(out); in logsumexp_out()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor.cpp417 return at::functionalization::impl::to_functional_tensor(out); in to_lazy_tensor()
/aosp_15_r20/external/pytorch/torch/csrc/functorch/
H A Dinit.cpp43 auto t = at::functionalization::impl::to_functional_tensor(self); in _wrap_functional_tensor()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_torch_functions_manual.cpp751 return at::functionalization::impl::to_functional_tensor(t); in initTorchFunctions()