Home
last modified time | relevance | path

Searched defs:ZeroTensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DZeroTensorFallback.cpp86 TORCH_LIBRARY_IMPL(_, ZeroTensor, m) { in TORCH_LIBRARY_IMPL() argument
90 TORCH_LIBRARY_IMPL(aten, ZeroTensor, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKey.h273 ZeroTensor, // registered at build/aten/src/ATen/RegisterZeroTensor.cpp enumerator
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py103 ZeroTensor = auto() variable in DispatchKey
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx.py1466 class ZeroTensor(metaclass=torch.fx.ProxyableClassMeta): class