Searched refs:isFunctionalTensor (Results 1 – 10 of 10) sorted by relevance
59 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(value_)); in FunctionalTensorWrapper()143 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(value_)); in FunctionalTensorWrapper()221 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(other)); in replace_()387 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(t)); in regenerate_from_base()389 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(t)); in regenerate_from_base()529 TORCH_INTERNAL_ASSERT_DEBUG_ONLY(!isFunctionalTensor(tensor)); in to_functional_tensor()560 if (isFunctionalTensor(tensor)) { in from_functional_tensor()610 if (!at::functionalization::impl::isFunctionalTensor(t)) { in sync()633 TORCH_INTERNAL_ASSERT_DEBUG_ONLY(isFunctionalTensor(functional_tensor)); in replace_()647 TORCH_INTERNAL_ASSERT_DEBUG_ONLY(isFunctionalTensor(functional_tensor)); in propagate_xla_data()[all …]
67 if (t.defined() && at::functionalization::impl::isFunctionalTensor(t)) { in functionalizeFallback()76 if (at::functionalization::impl::isFunctionalTensor(tensors)) { in functionalizeFallback()85 if (at::functionalization::impl::isFunctionalTensor(opt_tensors)) { in functionalizeFallback()131 if (at::functionalization::impl::isFunctionalTensor(self)) { in resize__functionalization()138 if (!at::functionalization::impl::isFunctionalTensor(self)) { in resize__functionalization()190 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(self)); in lift_functionalize()198 if (at::functionalization::impl::isFunctionalTensor(self)) { in lift_fresh_functionalize()213 if (at::functionalization::impl::isFunctionalTensor(self)) { in lift_fresh_functionalize_copy()245 if (at::functionalization::impl::isFunctionalTensor(self)) { in _to_copy_functionalize()290 if (!at::functionalization::impl::isFunctionalTensor(self)) { in _unsafe_view_functionalize()[all …]
44 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(t)); in apply_update()62 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(t)); in apply_update()111 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(base_));
301 TORCH_API bool isFunctionalTensor(const at::Tensor& tensor);302 TORCH_API bool isFunctionalTensor(const std::optional<Tensor>& t);303 TORCH_API bool isFunctionalTensor(305 TORCH_API bool isFunctionalTensor(ITensorListRef list);
611 at::functionalization::impl::isFunctionalTensor(t)); in initTorchFunctions()618 at::functionalization::impl::isFunctionalTensor(t)); in initTorchFunctions()626 at::functionalization::impl::isFunctionalTensor(t)); in initTorchFunctions()634 at::functionalization::impl::isFunctionalTensor(t)); in initTorchFunctions()641 at::functionalization::impl::isFunctionalTensor(tensor)); in initTorchFunctions()647 TORCH_INTERNAL_ASSERT(at::functionalization::impl::isFunctionalTensor(t)); in initTorchFunctions()652 TORCH_INTERNAL_ASSERT(at::functionalization::impl::isFunctionalTensor(t)); in initTorchFunctions()656 TORCH_INTERNAL_ASSERT(at::functionalization::impl::isFunctionalTensor(t)); in initTorchFunctions()662 at::functionalization::impl::isFunctionalTensor(t)); in initTorchFunctions()664 !at::functionalization::impl::isFunctionalTensor(o)); in initTorchFunctions()[all …]
10 TORCH_INTERNAL_ASSERT(!at::functionalization::impl::isFunctionalTensor(tensor)); in sanityCheckNotFunctional()33 if (at::functionalization::impl::isFunctionalTensor(tensor)) { in processImpl()
52 at::functionalization::impl::isFunctionalTensor(wrapped)); in _assert_wrapped_functional()54 !at::functionalization::impl::isFunctionalTensor(unwrapped)); in _assert_wrapped_functional()66 at::functionalization::impl::isFunctionalTensor(wrapped)); in _propagate_functional_input_mutation()68 !at::functionalization::impl::isFunctionalTensor(unwrapped)); in _propagate_functional_input_mutation()174 TORCH_INTERNAL_ASSERT(at::functionalization::impl::isFunctionalTensor(self)); in _unwrap_functional_tensor()331 if (at::functionalization::impl::isFunctionalTensor(tensor)) { in get_unwrapped()353 if (at::functionalization::impl::isFunctionalTensor(tensor)) { in maybe_get_level()
36 if (!at::functionalization::impl::isFunctionalTensor(variable)) { in resize_storage_bytes__functionalize()
69 if (at::functionalization::impl::isFunctionalTensor(tensor)) { in maybe_unwrap_functional()
389 !at::functionalization::impl::isFunctionalTensor(tensor)); in lift()394 !at::functionalization::impl::isFunctionalTensor(tensor)); in lift_fresh()