Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_device.cpp123 at::Tensor custom_empty_memory_format(at::IntArrayRef size, in custom_empty_memory_format() function
160 m.impl("empty.memory_format", &custom_empty_memory_format); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp287 at::Tensor custom_empty_memory_format(at::IntArrayRef size, in custom_empty_memory_format() function