Searched refs:to_functional_tensor (Results 1 – 7 of 7) sorted by relevance
109 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()
524 Tensor to_functional_tensor(const Tensor& tensor) { in to_functional_tensor() function532 std::optional<Tensor> to_functional_tensor(const std::optional<Tensor>& tensor) { in to_functional_tensor() function534 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() function542 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() function550 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()
307 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);
297 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()
417 return at::functionalization::impl::to_functional_tensor(out); in to_lazy_tensor()
43 auto t = at::functionalization::impl::to_functional_tensor(self); in _wrap_functional_tensor()
751 return at::functionalization::impl::to_functional_tensor(t); in initTorchFunctions()