Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A Dbackend_fallback_test.cpp34 struct GenericWrapperTensorImpl : public c10::TensorImpl { struct
35 explicit GenericWrapperTensorImpl(at::Tensor rep) in GenericWrapperTensorImpl() function
60 auto* wrapper = static_cast<GenericWrapperTensorImpl*>(impl); in generic_wrapper_fallback()
77 …torch::jit::push(*stack, at::detail::make_tensor<GenericWrapperTensorImpl>(std::move(rets[i]).toTe… in generic_wrapper_fallback()
102 Tensor a = at::detail::make_tensor<GenericWrapperTensorImpl>(ones({5, 5}, kDouble)); in TEST()