Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DRegisterOpContextClass.cpp35 -> SerializationTypeConv2dPrePack { // __getstate__ in TORCH_LIBRARY()
38 [](SerializationTypeConv2dPrePack state) in TORCH_LIBRARY()
H A DOpContext.h16 using SerializationTypeConv2dPrePack = std::tuple< variable
98 SerializationTypeConv2dPrePack unpack() { in unpack()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/
H A DMetalPrepackOpRegister.cpp46 -> SerializationTypeConv2dPrePack { // __getstate__ in TORCH_LIBRARY()
49 [](SerializationTypeConv2dPrePack state) in TORCH_LIBRARY()
H A DMetalPrepackOpContext.h8 using SerializationTypeConv2dPrePack = std::tuple< variable
20 SerializationTypeConv2dPrePack pack() { in pack()