vkml 0.0.2

High-level Vulkan-based machine learning library
1
2
3
4
5
6
7
8
9
10

use std::collections::HashMap;

use crate::{instruction::Instruction, tensor::TensorDesc, tensor_graph::TensorId};

pub struct LayerExecution {
    pub tensors: Vec<TensorDesc>,                // Tensor descriptors
    pub instructions: Vec<Box<dyn Instruction>>, // Layer-local instructions
    pub outputs: Vec<TensorId>,                  // Output tensor IDs
    pub input_mappings: HashMap<TensorId, (usize, TensorId)>, // Maps local tensor to (input_connection_idx, output_idx)
}