1 #pragma once 2 3 #include <torch/csrc/api/include/torch/jit.h> 4 #include <torch/csrc/lazy/backend/lowering_context.h> 5 6 namespace torch { 7 namespace lazy { 8 using TSOpVector = std::vector<torch::jit::Value*>; 9 10 TORCH_API TSOpVector LowerTSBuiltin( 11 std::shared_ptr<torch::jit::GraphFunction> function, 12 c10::Symbol sym, 13 const std::vector<torch::jit::NamedValue>& arguments, 14 const std::vector<torch::jit::NamedValue>& kwarguments = {}); 15 16 } // namespace lazy 17 } // namespace torch 18