vkml 0.0.3

High-level Vulkan-based machine learning library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

pub type LayerId = usize;

#[derive(Clone, Debug, PartialEq)]
pub enum LayerConnection {
    DefaultOutput(LayerId),
    SpecificOutput(LayerId, usize),
}

impl LayerConnection {
    pub fn get_layerid(&self) -> LayerId {
        match self {
            LayerConnection::DefaultOutput(id) => *id,
            LayerConnection::SpecificOutput(id, _) => *id,
        }
    }

    pub fn get_outputidx(&self) -> usize {
        match self {
            LayerConnection::DefaultOutput(_) => 0,
            LayerConnection::SpecificOutput(_, idx) => *idx,
        }
    }
}