Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_backend.cpp36 auto lm = torch::jit::detail::codegen_backend_module( in TEST()
108 auto lm = torch::jit::detail::codegen_backend_module( in TEST()
134 auto lm = torch::jit::detail::codegen_backend_module( in TEST()
165 auto lm = torch::jit::detail::codegen_backend_module( in TEST()
190 auto lm_add = torch::jit::detail::codegen_backend_module( in TEST()
198 auto lm_sub = torch::jit::detail::codegen_backend_module( in TEST()
258 auto lowered_sub1 = torch::jit::detail::codegen_backend_module( in getCompositeModuleWithSameNameSubModules()
266 auto lowered_sub2 = torch::jit::detail::codegen_backend_module( in getCompositeModuleWithSameNameSubModules()
366 torch::jit::detail::codegen_backend_module( in TEST()
388 auto lm = torch::jit::detail::codegen_backend_module( in TEST()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/
H A Dbackend_detail.h34 TORCH_API Module codegen_backend_module(
H A Dbackend_init.cpp128 return detail::codegen_backend_module( in codegen_func()
H A Dbackend_detail.cpp97 Module codegen_backend_module( in codegen_backend_module() function
/aosp_15_r20/external/pytorch/test/mobile/nnc/
H A Dtest_nnc_backend.cpp88 auto compiled_module = torch::jit::detail::codegen_backend_module( in TEST()
/aosp_15_r20/external/pytorch/binaries/
H A Daot_model_compiler.cc135 auto compiled_module = torch::jit::detail::codegen_backend_module( in main()