Home
last modified time | relevance | path

Searched refs:compileKernel (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dcompiler.h24 TORCH_API std::shared_ptr<FusedKernel> compileKernel(
H A Dcompiler.cpp200 std::shared_ptr<FusedKernel> compileKernel( in compileKernel() function
H A Dexecutor.cpp381 const auto kernel = compileKernel(spec, arg_spec, *maybe_map_size, device); in runFusion()