Function torch::jit::compile

• Defined in File jit.h

Function Documentation

std::shared_ptr<CompilationUnit> torch::jit::compile(const std::string &source)

Compiles script code into an executable graph.

Takes a string containing functions in script syntax and compiles them into a module (graph). The returned module provides a run_method function that may be used to invoke the compiled functions.

For example:

auto module = torch::jit::compile(R"JIT(
  def relu_script(a, b):
    return torch.relu(a + b)
  def test_while(a, i):
    while i < 10:
      a += a
      i += 1
    return a
)JIT");
IValue output = module->run_method("relu_script", a, b);