Packages

extorch

0.1.0-pre0
0.4.0 0.3.0 0.2.0 0.1.0-pre0

Production ML model serving on the BEAM. Serve PyTorch models faster than Python with pre-compiled graph execution, AOTI compiled inference, and OTP fault tolerance.

HexDocs

Current section

Files

Jump to
Readme
4 Versions
1 Dependency
Dependants
Files
Activity
extorch native extorch include common.h
Raw

native/extorch/include/common.h 

#pragma once

#include <memory>

#include <iostream>

#include <torch/torch.h>

#include "rust/cxx.h"



struct Device;

struct Scalar;

struct ScalarList;

struct TorchSlice;

struct TorchIndex;

using CrossTensor = torch::Tensor;