tensorflow-sys 0.3.1

//! Binding to [TensorFlow][1].
//!
//! [1]: https://www.tensorflow.org

#![allow(non_camel_case_types)]

extern crate libc;

use libc::{c_char, c_int, c_longlong, c_void, size_t};

#[repr(C)]
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum TF_DataType {
    TF_FLOAT = 1,
    TF_DOUBLE = 2,
    TF_INT32 = 3,
    TF_UINT8 = 4,
    TF_INT16 = 5,
    TF_INT8 = 6,
    TF_STRING = 7,
    TF_COMPLEX64 = 8,
    TF_INT64 = 9,
    TF_BOOL = 10,
    TF_QINT8 = 11,
    TF_QUINT8 = 12,
    TF_QINT32 = 13,
    TF_BFLOAT16 = 14,
    TF_QINT16 = 15,
    TF_QUINT16 = 16,
    TF_UINT16 = 17,
    TF_COMPLEX128 = 18,
    TF_HALF = 19,
}
pub use TF_DataType::*;

#[repr(C)]
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum TF_Code {
    TF_OK = 0,
    TF_CANCELLED = 1,
    TF_UNKNOWN = 2,
    TF_INVALID_ARGUMENT = 3,
    TF_DEADLINE_EXCEEDED = 4,
    TF_NOT_FOUND = 5,
    TF_ALREADY_EXISTS = 6,
    TF_PERMISSION_DENIED = 7,
    TF_UNAUTHENTICATED = 16,
    TF_RESOURCE_EXHAUSTED = 8,
    TF_FAILED_PRECONDITION = 9,
    TF_ABORTED = 10,
    TF_OUT_OF_RANGE = 11,
    TF_UNIMPLEMENTED = 12,
    TF_INTERNAL = 13,
    TF_UNAVAILABLE = 14,
    TF_DATA_LOSS = 15,
}
pub use TF_Code::*;

#[derive(Clone, Copy, Debug)]
pub enum TF_Status {}

#[repr(C)]
#[derive(Clone, Copy, Debug)]
pub struct TF_Buffer {
  pub data: *const c_void,
  pub length: size_t,
  pub data_deallocator: Option<unsafe extern "C" fn(data: *mut c_void, length: size_t)>,
}

#[derive(Clone, Copy, Debug)]
pub enum TF_Library {}

#[derive(Clone, Copy, Debug)]
pub enum TF_Tensor {}

#[derive(Clone, Copy, Debug)]
pub enum TF_SessionOptions {}

#[derive(Clone, Copy, Debug)]
pub enum TF_Session {}

extern "C" {
    pub fn TF_NewBufferFromString(proto: *const c_void, proto_len: size_t) -> *mut TF_Buffer;

    pub fn TF_NewBuffer() -> *mut TF_Buffer;

    pub fn TF_DeleteBuffer(buffer: *mut TF_Buffer);

    pub fn TF_GetBuffer(buffer: *mut TF_Buffer) -> TF_Buffer;

    pub fn TF_NewStatus() -> *mut TF_Status;

    pub fn TF_DeleteStatus(status: *mut TF_Status);

    pub fn TF_SetStatus(status: *mut TF_Status, code: TF_Code, msg: *const c_char);

    pub fn TF_GetCode(status: *const TF_Status) -> TF_Code;

    pub fn TF_Message(status: *const TF_Status) -> *const c_char;

    pub fn TF_NewTensor(datatype: TF_DataType, dims: *mut c_longlong, num_dims: c_int,
                        data: *mut c_void, len: size_t,
                        deallocator: Option<unsafe extern "C" fn(data: *mut c_void, len: size_t,
                                                                 arg: *mut c_void)>,
                        deallocator_arg: *mut c_void) -> *mut TF_Tensor;

    pub fn TF_DeleteTensor(tensor: *mut TF_Tensor);

    pub fn TF_TensorType(tensor: *const TF_Tensor) -> TF_DataType;

    pub fn TF_NumDims(tensor: *const TF_Tensor) -> c_int;

    pub fn TF_Dim(tensor: *const TF_Tensor, dim_index: c_int) -> c_longlong;

    pub fn TF_TensorByteSize(tensor: *const TF_Tensor) -> size_t;

    pub fn TF_TensorData(tensor: *const TF_Tensor) -> *mut c_void;

    pub fn TF_NewSessionOptions() -> *mut TF_SessionOptions;

    pub fn TF_SetTarget(options: *mut TF_SessionOptions, target: *const c_char);

    pub fn TF_SetConfig(options: *mut TF_SessionOptions, proto: *const c_void, proto_len: size_t,
                        status: *mut TF_Status);

    pub fn TF_DeleteSessionOptions(options: *mut TF_SessionOptions);

    pub fn TF_NewSession(options: *const TF_SessionOptions, status: *mut TF_Status)
                         -> *mut TF_Session;

    pub fn TF_CloseSession(session: *mut TF_Session, status: *mut TF_Status);

    pub fn TF_DeleteSession(session: *mut TF_Session, status: *mut TF_Status);

    pub fn TF_ExtendGraph(session: *mut TF_Session, proto: *const c_void, proto_len: size_t,
                          status: *mut TF_Status);

    pub fn TF_Run(session: *mut TF_Session, run_options: *const TF_Buffer,
                  input_names: *mut *const c_char, inputs: *mut *mut TF_Tensor, ninputs: c_int,
                  output_tensor_names: *mut *const c_char, outputs: *mut *mut TF_Tensor,
                  noutputs: c_int, target_node_names: *mut *const c_char, ntargets: c_int,
                  run_metadata: *mut TF_Buffer, status: *mut TF_Status);

    pub fn TF_PRunSetup(session: *mut TF_Session, input_names: *mut *const c_char, ninputs: c_int,
                        output_tensor_names: *mut *const c_char, noutputs: c_int,
                        target_node_names: *mut *const c_char, ntargets: c_int,
                        handle: *mut *mut c_char, status: *mut TF_Status);

    pub fn TF_PRun(session: *mut TF_Session, handle: *const c_char,
                   input_names: *mut *const c_char, inputs: *mut *mut TF_Tensor, ninputs: c_int,
                   output_tensor_names: *mut *const c_char, outputs: *mut *mut TF_Tensor,
                   noutputs: c_int, target_node_names: *mut *const c_char, ntargets: c_int,
                   status: *mut TF_Status);

    pub fn TF_LoadLibrary(library_filename: *const c_char, status: *mut TF_Status)
                          -> *mut TF_Library;

    pub fn TF_GetOpList(lib_handle: *mut TF_Library) -> TF_Buffer;
}