Trait tensorflow::TensorType

pub trait TensorType: Default + Clone + Display + Debug + 'static {
    // Required methods
    fn data_type() -> DataType;
    fn zero() -> Self;
    fn one() -> Self;
    fn is_repr_c() -> bool;
    fn unpack(data: &[u8], count: usize) -> Result<Vec<Self>>;
    fn pack(data: &[Self], dims: &[u64]) -> Result<*mut TF_Tensor>;
}

A Rust type that maps to a DataType.

Currently, all implementors must not implement Drop (or transitively contain anything that does) and must be bit-for-bit compatible with the corresponding C type. Clients must not implement this trait.

This trait doesn’t require num::Zero or num::One because some tensor types (such as bool and String) don’t implement them and we need to supply custom implementations.

Required Methods

fn data_type() -> DataType

Returns the DataType that corresponds to this type.

fn zero() -> Self

Returns the zero value.

fn one() -> Self

Returns the one value.

fn is_repr_c() -> bool

Return true if the data has the same representation in C and Rust and can be written/read directly.

fn unpack(data: &[u8], count: usize) -> Result<Vec<Self>>

Unpacks data from C. Returns an error if is_repr_c() is true for this type or some other error occurred.

fn pack(data: &[Self], dims: &[u64]) -> Result<*mut TF_Tensor>

Packs data for sending to C. Returns an error if is_repr_c() returns true for this type or some other error occurred.

Implementations on Foreign Types

impl TensorType for u8

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for u64

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for f32

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for i32

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for Complex<f64>

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for f16

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for u32

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for i64

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for Complex<f32>

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for String

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(data: &[u8], count: usize) -> Result<Vec<Self>>

fn pack(data: &[Self], dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for f64

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for i8

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for bool

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for i16

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

impl TensorType for u16

fn data_type() -> DataType

fn zero() -> Self

fn one() -> Self

fn is_repr_c() -> bool

fn unpack(_data: &[u8], _count: usize) -> Result<Vec<Self>>

fn pack(_data: &[Self], _dims: &[u64]) -> Result<*mut TF_Tensor>

Implementors

impl TensorType for BFloat16

impl TensorType for QInt8

impl TensorType for QInt16

impl TensorType for QInt32

impl TensorType for QUInt8

impl TensorType for QUInt16