Searched defs:method_compile_spec (Results 1 – 14 of 14) sorted by relevance
/aosp_15_r20/external/pytorch/test/custom_backend/ |
H A D | custom_backend.h | 23 c10::impl::GenericDict method_compile_spec) override { in compile() 72 const c10::Dict<c10::IValue, c10::IValue>& method_compile_spec, in preprocess()
|
/aosp_15_r20/external/pytorch/test/cpp/jit/ |
H A D | test_backend_lib.cpp | 25 c10::impl::GenericDict method_compile_spec) override { in compile() 75 const c10::Dict<IValue, IValue>& method_compile_spec, in preprocess() argument
|
H A D | test_backend_compiler_preprocess.cpp | 15 const c10::Dict<IValue, IValue>& method_compile_spec, in preprocess() argument
|
H A D | test_backend_compiler_lib.cpp | 86 c10::impl::GenericDict method_compile_spec) override { in compile()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/ |
H A D | backend_init.cpp | 125 const py::dict& method_compile_spec) { in codegen_func() 150 const py::dict& method_compile_spec) { in initJitBackendBindings()
|
H A D | backend_detail.cpp | 100 const c10::Dict<IValue, IValue>& method_compile_spec, in codegen_backend_module() argument
|
/aosp_15_r20/external/pytorch/test/jit/ |
H A D | test_backends.py | 36 def to_test_backend(module, method_compile_spec): argument 42 def to_test_backend_multi(module, method_compile_spec): argument 46 def to_test_backend_selective(module, method_compile_spec, submodules): argument
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/coreml/cpp/ |
H A D | backend.cpp | 10 c10::impl::GenericDict method_compile_spec) override { in compile()
|
H A D | preprocess.cpp | 19 const c10::Dict<c10::IValue, c10::IValue>& method_compile_spec, in preprocess()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/ |
H A D | backend.cpp | 22 c10::impl::GenericDict method_compile_spec) override { in compile()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/nnapi/ |
H A D | nnapi_backend_preprocess.cpp | 31 const c10::Dict<c10::IValue, c10::IValue>& method_compile_spec, in preprocess()
|
H A D | nnapi_backend_lib.cpp | 33 c10::impl::GenericDict method_compile_spec) override { in compile()
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/ |
H A D | xnnpack_backend_preprocess.cpp | 33 const c10::Dict<c10::IValue, c10::IValue>& method_compile_spec, in preprocess()
|
H A D | xnnpack_backend_lib.cpp | 38 c10::impl::GenericDict method_compile_spec) override { in compile()
|