moose 0.2.2

Encrypted learning and data processing 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

//! Abstraction layer for floating-point values.

use crate::computation::{Placed, Placement};
use crate::error::Result;
use serde::{Deserialize, Serialize};

mod ops;

#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub enum FloatTensor<HostTenT, Mir3TenT> {
    Host(HostTenT),
    Mirrored3(Mir3TenT),
}

impl<HostTenT, Mir3TenT> Placed for FloatTensor<HostTenT, Mir3TenT>
where
    HostTenT: Placed,
    HostTenT::Placement: Into<Placement>,
    Mir3TenT: Placed,
    Mir3TenT::Placement: Into<Placement>,
{
    type Placement = Placement;

    fn placement(&self) -> Result<Self::Placement> {
        match self {
            FloatTensor::Host(x) => Ok(x.placement()?.into()),
            FloatTensor::Mirrored3(x) => Ok(x.placement()?.into()),
        }
    }
}