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
31
32

//! Abstraction layer for integer values.

use crate::computation::*;
use crate::error::Result;
use crate::kernels::*;
use serde::{Deserialize, Serialize};

mod ops;

/// Uint64 Tensor abstracting over host and replicated values
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub enum AbstractUint64Tensor<HostT, RepT> {
    Host(HostT),
    Replicated(RepT),
}

impl<HostT, RepT> Placed for AbstractUint64Tensor<HostT, RepT>
where
    HostT: Placed,
    HostT::Placement: Into<Placement>,
    RepT: Placed,
    RepT::Placement: Into<Placement>,
{
    type Placement = Placement;

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