Home
last modified time | relevance | path

Searched refs:from_functional_tensor (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/aten/src/ATen/
DFunctionalTensorWrapper.cpp555 Tensor from_functional_tensor(const Tensor& tensor, bool assert_functional) { in from_functional_tensor() function
570 std::optional<Tensor> from_functional_tensor(const std::optional<Tensor>& t, bool assert_functional… in from_functional_tensor() function
572 return std::make_optional<Tensor>(from_functional_tensor(*t, assert_functional)); in from_functional_tensor()
576 std::vector<Tensor> from_functional_tensor(ITensorListRef t_list) { in from_functional_tensor() function
586 outputs.push_back(from_functional_tensor(tensor, /*assert_functional=*/false)); in from_functional_tensor()
590 c10::List<::std::optional<Tensor>> from_functional_tensor(const c10::List<::std::optional<Tensor>>&… in from_functional_tensor() function
594 outputs.push_back(from_functional_tensor(t_list[i], /*assert_functional=*/false)); in from_functional_tensor()
885 auto t_new = c10::IValue(at::functionalization::impl::from_functional_tensor(t)); in functionalize_op_helper()
890 auto t_new = c10::IValue(at::functionalization::impl::from_functional_tensor(tensors)); in functionalize_op_helper()
895 auto t_new = c10::IValue(at::functionalization::impl::from_functional_tensor(opt_tensors)); in functionalize_op_helper()
DFunctionalizeFallbackKernel.cpp70 auto t_new = c10::IValue(at::functionalization::impl::from_functional_tensor(t)); in functionalizeFallback()
79 auto t_new = c10::IValue(at::functionalization::impl::from_functional_tensor(tensors)); in functionalizeFallback()
88 … auto t_new = c10::IValue(at::functionalization::impl::from_functional_tensor(opt_tensors)); in functionalizeFallback()
133 self_ = at::functionalization::impl::from_functional_tensor(self); in resize__functionalization()
249 self_ = at::functionalization::impl::from_functional_tensor(self); in _to_copy_functionalize()
295 auto self_ = at::functionalization::impl::from_functional_tensor(self); in _unsafe_view_functionalize()
DFunctionalTensorWrapper.h317 from_functional_tensor(const Tensor& tensor, bool assert_functional = true);
318 TORCH_API std::optional<Tensor> from_functional_tensor(
321 TORCH_API c10::List<std::optional<Tensor>> from_functional_tensor(
323 TORCH_API std::vector<Tensor> from_functional_tensor(ITensorListRef t_list);
/external/pytorch/torch/csrc/lazy/python/
Dinit.cpp50 auto inner = at::functionalization::impl::from_functional_tensor(tensor); in GetTensorsDump()
/external/pytorch/torch/csrc/lazy/ts_backend/
Dts_native_functions.cpp488 at::functionalization::impl::from_functional_tensor(out_wrapped); in logsumexp_out()
/external/pytorch/torch/csrc/autograd/
Dpython_torch_functions_manual.cpp754 return at::functionalization::impl::from_functional_tensor(t); in initTorchFunctions()