Home
last modified time | relevance | path

Searched refs:is_python_dispatch (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dinput_metadata.cpp11 if (input.is_nested() && !input.unsafeGetTensorImpl()->is_python_dispatch()) { in compute_variant_shape()
18 bool is_python_dispatch(const at::Tensor& tensor) { in is_python_dispatch() function
19 return tensor.unsafeGetTensorImpl()->is_python_dispatch(); in is_python_dispatch()
23 return tensor.is_nested() && !is_python_dispatch(tensor); in is_cpp_nested_tensor()
46 is_python_dispatch(t), in InputMetadata()
199 grad_is_nested && is_python_dispatch(grad) && in maybe_expandable_to()
H A Dvariable.cpp716 self.unsafeGetTensorImpl()->is_python_dispatch(), in grad_fn()
H A Dengine.cpp925 grad.unsafeGetTensorImpl()->is_python_dispatch())) { in validate_outputs()
H A Dpython_variable.cpp897 var.unsafeGetTensorImpl()->is_python_dispatch()); in THPVariable_get_python_dispatch()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_list.cpp47 !tensor.unsafeGetTensorImpl()->is_python_dispatch(), in tensor_to_list()
H A Dtensor_numpy.cpp130 !tensor.unsafeGetTensorImpl()->is_python_dispatch(), in tensor_to_numpy()
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp911 …!self_.unsafeGetTensorImpl()->is_python_dispatch() && !other.unsafeGetTensorImpl()->is_python_disp… in THPVariable_map_()
940 … !x.unsafeGetTensorImpl()->is_python_dispatch() && !y.unsafeGetTensorImpl()->is_python_dispatch(), in THPVariable_map2_()
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.h951 TORCH_INTERNAL_ASSERT(is_python_dispatch()) in matches_python_custom()
1956 bool is_python_dispatch() const { in is_python_dispatch() function