openinfer-dsl 0.1.0

Rust-embedded DSL for defining OpenInfer graphs with explicit control flow and memory semantics.
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

use proc_macro2::TokenStream;
use quote::quote;
use syn::Ident;

use crate::types::{OpAttrValue, OpSetting};

use super::{attr_value_expr, SettingsMap};

pub(crate) fn build_attrs(op: &Ident, settings: &[OpSetting]) -> syn::Result<TokenStream> {
    let mut settings = SettingsMap::new(op, settings)?;

    let alpha =
        settings
            .take_value("alpha")
            .unwrap_or_else(|| OpAttrValue::Double(0.0));
    let clamp_max =
        settings
            .take_value("clamp_max")
            .unwrap_or_else(|| OpAttrValue::Double(f64::INFINITY));

    settings.ensure_empty()?;

    let alpha_expr = attr_value_expr(&alpha);
    let clamp_max_expr = attr_value_expr(&clamp_max);

    Ok(quote! {
        ::openinfer::OpAttrs::Relu {
            alpha: #alpha_expr,
            clamp_max: #clamp_max_expr,
        }
    })
}