Home
last modified time | relevance | path

Searched defs:opt_memory_format (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp441 auto opt_memory_format = r.memoryformatOptional(0); in THPVariable_cpu() local
497 auto opt_memory_format = r.memoryformatOptional(2); in THPVariable_cuda() local
520 auto opt_memory_format = r.memoryformatOptional(2); in THPVariable_mtia() local
543 auto opt_memory_format = r.memoryformatOptional(2); in THPVariable_xpu() local
566 auto opt_memory_format = r.memoryformatOptional(2); in THPVariable_ipu() local
591 auto opt_memory_format = r.memoryformatOptional(0); in THPVariable_byte() local
608 auto opt_memory_format = r.memoryformatOptional(0); in THPVariable_char() local
625 auto opt_memory_format = r.memoryformatOptional(0); in THPVariable_double() local
642 auto opt_memory_format = r.memoryformatOptional(0); in THPVariable_float() local
659 auto opt_memory_format = r.memoryformatOptional(0); in THPVariable_cdouble() local
[all …]
H A Dpython_nn_functions.cpp53 auto opt_memory_format = std::get<4>(parsed); in THPVariable__parse_to() local