Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DPythonFallbackKernel.cpp73 auto* interpreter = ivalue.unsafeToTensorImpl()->pyobj_slot()->pyobj_interpreter(); in pythonFallback()
85 auto* interpreter = nv.unsafeToTensorImpl()->pyobj_slot()->pyobj_interpreter(); in pythonFallback()
H A Divalue.h440 at::TensorImpl* unsafeToTensorImpl() const { in unsafeToTensorImpl() function
1393 payload.as_intrusive_ptr = rhs.unsafeToTensorImpl(); in WeakIValue()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp151 ivalue.unsafeToTensorImpl()->pyobj_slot()->pyobj_interpreter(); in operator ()()
153 ivalue.unsafeToTensorImpl()->key_set().has( in operator ()()
168 nv.unsafeToTensorImpl()->pyobj_slot()->pyobj_interpreter(); in operator ()()
170 nv.unsafeToTensorImpl()->key_set().has(at::DispatchKey::Python)) { in operator ()()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A Dbackend_fallback_test.cpp58 auto* impl = args[i].unsafeToTensorImpl(); in generic_wrapper_fallback()
H A DDispatchKeyExtractor.h152 ks = ks | ivalue.unsafeToTensorImpl()->key_set(); in getDispatchKeySetBoxed()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/
H A Dserialize.h273 ivalue.unsafeToTensorImpl() == \
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Divalue_test.cpp616 EXPECT_EQ(tensor.internalToPointer(), tensor.unsafeToTensorImpl()); in TEST()