Struct FlexibleSession 

pub struct FlexibleSession { /* private fields */ }

Compile-time settings without a fixed Device.

Pick the backend per graph via Self::compile_resolved or Self::compile_on.

Implementations

impl FlexibleSession

pub fn new() -> Self

pub fn from_env() -> Self

pub fn with_device_policy(self, policy: DevicePolicy) -> Self

pub fn with_precision(self, precision: Precision) -> Self

pub fn with_op_policy(self, policy: PrecisionPolicy) -> Self

pub fn device_policy(&self) -> &DevicePolicy

pub fn precision(&self) -> Precision

pub fn compile_on( &self, graph: Graph, device: Device, ) -> Result<CompiledGraph, String>

pub fn compile_with_on( &self, graph: Graph, device: Device, options: &CompileOptions, ) -> Result<CompiledGraph, String>

pub fn compile_resolved( &self, graph: Graph, hint: Option<Device>, ) -> Result<CompiledGraph, String>

pub fn compile_resolved_with( &self, graph: Graph, hint: Option<Device>, options: &CompileOptions, ) -> Result<CompiledGraph, String>

impl FlexibleSession

pub fn compile_hir_on( &self, hir: HirModule, device: Device, ) -> Result<CompiledGraph, LowerError>

pub fn compile_module_on( &self, module: GraphModule, device: Device, ) -> Result<CompiledGraph, LowerError>

Trait Implementations

impl Default for FlexibleSession

fn default() -> Self

Returns the “default value” for a type.