This crate provides Rust bindings for the
TensorFlow machine learning library.
If you aren't sure how to use something, please see the examples folder.
This module builds computation graphs.
A module for writing TFRecords, Tensorflow's preferred on-disk data format.
AttrMetadata describes the value of an attribute on an operation.
BFloat16 provides a Rust type for BFloat16.
Metadata about a device.
An opaque token for retrieving an output from a computation.
Function is a grouping of operations with defined inputs and outputs. Once created and added to graphs, functions can be invoked by creating an operation whose operation type matches the function name.
Options that can be passed during function creation.
Represents a computation graph. Graphs may be shared between sessions. Graphs are thread-safe when used as directed.
ImportGraphDefResults holds results that are generated by Graph::import_graph_def_with_results().
Dynamically loaded plugins. The C API doesn't provide a way to unload libraries, so nothing happens when this goes out of scope.
Iterator over the operations in a
Quantized type for i8.
Quantized type for i16.
Quantized type for i32.
Quantized type for u8.
Quantized type for u16.
Aggregation type for a saved model bundle.
Manages a single graph and execution.
Options that can be passed during session creation.
Manages the inputs and outputs for a single execution of a graph.
A Shape is the shape of a tensor. A Shape may be an unknown rank, or it may have a known rank with each dimension being known or unknown.
Holds error information when communicating with back and forth with
Holds a multi-dimensional array of elements of a single data type.
A WhileBuilder is used to build a while loop.
Describes the type of the value of an attribute on an operation.
Error values that can be returned.
Type of a single tensor element.
A Rust type that maps to a
Returns a serialized KernelList protocol buffer containing KernelDefs for all registered kernels.
Returns a serialized KernelList protocol buffer containing KernelDefs for
all kernels registered for the operation named
Returns a string describing version information of the
Deprecated alias for FetchToken.
Convenience type for
Deprecated alias for SessionRunArgs.