tenflowers-core 0.1.1

Core tensor operations and execution engine for TenfloweRS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

//! Device placement: strategies, manager, and graph optimizer.
//!
//! Submodules:
//! - `types`: Enums and structs (PlacementStrategy, OpInfo, PrecisionType, etc.)
//! - `manager`: DevicePlacement manager and global convenience functions
//! - `optimizer`: GraphPlacementOptimizer

pub mod manager;
pub mod optimizer;
pub mod types;

mod tests;

pub use manager::{
    choose_device_for_op, estimate_flops, estimate_memory_usage, get_placement_manager,
    set_placement_strategy, DevicePlacement,
};
pub use optimizer::GraphPlacementOptimizer;
pub use types::{
    CostWeights, DeviceCapabilities, GraphOpInfo, OpCategory, OpInfo, OptimizationStats,
    PlacementCost, PlacementStrategy, PrecisionType,
};