[−][src]Crate tensorflow
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.
Modules
| io | A module for writing TFRecords, Tensorflow's preferred on-disk data format. |
Structs
| AttrMetadata | AttrMetadata describes the value of an attribute on an operation. |
| BFloat16 | BFloat16 provides a Rust type for BFloat16. |
| Device | Metadata about a device. |
| FetchToken | An opaque token for retrieving an output from a computation. |
| Function | 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. |
| FunctionOptions | Options that can be passed during function creation. |
| Graph | Represents a computation graph. Graphs may be shared between sessions. Graphs are thread-safe when used as directed. |
| ImportGraphDefOptions |
|
| ImportGraphDefResults | ImportGraphDefResults holds results that are generated by Graph::import_graph_def_with_results(). |
| Input | A |
| Library | Dynamically loaded plugins. The C API doesn't provide a way to unload libraries, so nothing happens when this goes out of scope. |
| Operation | An |
| OperationDescription | An |
| OperationIter | Iterator over the operations in a |
| Output | A |
| QInt8 | Quantized type for i8. |
| QInt16 | Quantized type for i16. |
| QInt32 | Quantized type for i32. |
| QUInt8 | Quantized type for u8. |
| QUInt16 | Quantized type for u16. |
| SavedModelBundle | Aggregation type for a saved model bundle. |
| Session | Manages a single graph and execution. |
| SessionOptions | Options that can be passed during session creation. |
| SessionRunArgs | Manages the inputs and outputs for a single execution of a graph. |
| Shape | 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. |
| Status | Holds error information when communicating with back and forth with |
| Tensor | Holds a multi-dimensional array of elements of a single data type. |
| WhileBuilder | A WhileBuilder is used to build a while loop. |
Enums
| AttrType | Describes the type of the value of an attribute on an operation. |
| Code | Error values that can be returned. |
| DataType | Type of a single tensor element. |
Traits
| TensorType | A Rust type that maps to a |
Functions
| get_all_registered_kernels | Returns a serialized KernelList protocol buffer containing KernelDefs for all registered kernels. |
| get_registered_kernels_for_op | Returns a serialized KernelList protocol buffer containing KernelDefs for
all kernels registered for the operation named |
| version | Returns a string describing version information of the
|
Type Definitions
| OutputToken | Deprecated Deprecated alias for FetchToken. |
| Result | Convenience type for |
| StepWithGraph | Deprecated Deprecated alias for SessionRunArgs. |