tensorflow 0.21.0

Rust language bindings for TensorFlow.
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

#![allow(non_snake_case)]
/// Code for Op's ut that mimics raw_opw.
use crate::eager::{TensorHandle, ToTensorHandle};
use crate::Result;

use super::Op;

/// Add
#[derive(::std::fmt::Debug)]
pub struct Add {
    T: ::std::option::Option<crate::DataType>,
    device_name: ::std::option::Option<String>,
}

impl ::std::default::Default for Add {
    fn default() -> Self {
        Self {
            T: None,
            device_name: None,
        }
    }
}

impl Add {
    /// Creates a new `Add`.
    pub fn new() -> Self {
        Self::default()
    }

    /// Sets the `T` attribute.
    pub fn T<ArgType: ::std::convert::Into<crate::DataType>>(mut self, value: ArgType) -> Self {
        self.T = ::std::option::Option::Some(value.into());
        self
    }

    /// Set the `device_name` where in the Op is executed.
    pub fn set_device(&mut self, device_name: &str) {
        self.device_name = ::std::option::Option::Some(device_name.to_string());
    }

    /// Execute add.
    pub fn call<'a, T0, T1>(
        &self,
        ctx: &'a crate::eager::Context,
        x: &T0,
        y: &T1,
    ) -> Result<TensorHandle<'a>>
    where
        T0: ToTensorHandle<'a>,
        T1: ToTensorHandle<'a>,
    {
        // Define Op

        let op_name = "Add";
        let mut op = Op::new(ctx, op_name)?;

        // Required input arguments
        op.add_input(&x.to_handle(ctx)?)?;
        op.add_input(&y.to_handle(ctx)?)?;

        // Attributes
        if let ::std::option::Option::Some(value) = &self.T {
            let attr_name = "T";
            op.set_attr_type(attr_name, *value)?;
        }

        // Device
        if let ::std::option::Option::Some(device_name) = &self.device_name {
            op.set_device(device_name)?;
        }

        // Execute Op
        let [h] = op.execute::<1>(ctx)?;
        Ok(h)
    }
}

/// add with default options.
pub fn add<'a, T0, T1>(ctx: &'a crate::eager::Context, x: &T0, y: &T1) -> Result<TensorHandle<'a>>
where
    T0: ToTensorHandle<'a>,
    T1: ToTensorHandle<'a>,
{
    let op = Add::new();
    op.call(ctx, x, y)
}