burn-onnx 0.21.0-pre.3

Library for importing ONNX models into the Burn framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

/// Common imports for ONNX node implementations
///
/// This prelude provides all the commonly used types, traits, and functions
/// needed when implementing NodeCodegen for ONNX operators.
///
/// # Usage
///
/// Add this import at the top of your node implementation file:
///
/// ```ignore
/// use super::prelude::*;
/// ```
/// Re-export common traits and helpers from parent module
pub(crate) use super::broadcast_helpers;
pub(crate) use super::{NodeCodegen, arg_to_ident};
pub(crate) use crate::burn::argument_helpers::{
    on_device_to_native, scalar_native_to_shape, scalar_tensor_to_shape, shape_to_native,
};

// Re-export common burn-onnx types
pub(crate) use crate::burn::scope::ScopeAtPosition;
pub(crate) use crate::burn::{BurnImports, Field, Scope, ToTokens};

// Re-export common onnx_ir types
pub(crate) use onnx_ir::{
    Argument,
    ir::{ArgType, BoolStore, DType},
};

// Re-export common proc_macro2 types
pub(crate) use proc_macro2::{Ident, Span, TokenStream};

// Re-export quote macro
pub(crate) use quote::quote;