mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
bc68625151
Optimization Blocklist will be used in a future diff (D40315730) to make the rewrite to transfer input/output backends optional Differential Revision: [D40315729](https://our.internmc.facebook.com/intern/diff/D40315729/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/87431 Approved by: https://github.com/mcr229, https://github.com/digantdesai
19 lines
698 B
C++
19 lines
698 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/jit/api/module.h>
|
|
#include <torch/csrc/jit/ir/ir.h>
|
|
#include <torch/csrc/jit/passes/mobile_optimizer_type.h>
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
TORCH_API void vulkanInsertPrePackedOps(std::shared_ptr<Graph>& graph);
|
|
TORCH_API void vulkanInsertPrePackedOps(script::Module& module);
|
|
TORCH_API void vulkanFusePrePackedConvWithClamp(script::Module& module);
|
|
TORCH_API void vulkanFoldPrePackingOps(script::Module& module);
|
|
TORCH_API script::Module vulkanOptimizeForMobile(
|
|
const script::Module& module,
|
|
const std::set<MobileOptimizerType>& optimization_blocklist,
|
|
const std::vector<std::string>& preserved_methods);
|
|
} // namespace jit
|
|
} // namespace torch
|