Searched refs:compileKernel (Results 1 – 3 of 3) sorted by relevance
24 TORCH_API std::shared_ptr<FusedKernel> compileKernel(
200 std::shared_ptr<FusedKernel> compileKernel( in compileKernel() function
381 const auto kernel = compileKernel(spec, arg_spec, *maybe_map_size, device); in runFusion()