Struct caffe2_context::CPUContext

pub struct CPUContext { /* private fields */ }

| The CPU Context, representing the bare | minimum of what a Context class in Caffe2 | should implement. | | // TODO modify docs | | See operator.h, especially Operator, | for how Context are used in actual operator | implementations that are associated | with specific devices. | | In general, the Context class is passed | in as a template argument, and the operator | can use the functions defined in the | context to execute whatever computation | it has. |

Implementations

impl CPUContext

pub fn supports_non_fundamental_types(&self) -> bool

pub fn device(&self) -> Device

pub fn device_type(&self) -> DeviceType

pub fn get_device_type() -> DeviceType

pub fn new(nbytes: usize) -> DataPtr

pub fn copy_bytes_fromCPU( &mut self, nbytes: usize, src: *const c_void, dst: *mut c_void )

pub fn copy_bytes_toCPU( &mut self, nbytes: usize, src: *const c_void, dst: *mut c_void )

pub fn copy<T, SrcContext, DstContext>( &mut self, n: usize, src: *const T, dst: *mut T )

pub fn copy_items<SrcContext, DstContext>( &mut self, meta: TypeMeta, n: usize, src: *const c_void, dst: *mut c_void )

pub fn has_async_part_default() -> bool

By default CPU operators don’t have async device parts

pub fn supports_async_scheduling() -> bool

pub fn is_stream_free(option: &DeviceOption, stream_id: i32) -> bool

| CPU streams are not implemented and | are silently ignored by CPU ops, return | true to signal executor to schedule | a CPU op |

pub fn rand_generator(&mut self) -> *mut RandGenType

pub fn rand_seed(&mut self) -> u32

pub fn switch_to_device(&mut self, stream_id: i32)

pub fn wait_event(&mut self, ev: &Event)

pub fn record(&self, ev: *mut Event, err_msg: *const u8)

pub fn finish_device_computation(&mut self)

pub fn copy_bytes_same_device( &mut self, nbytes: usize, src: *const c_void, dst: *mut c_void )

pub fn copy_bytes(nbytes: usize, src: *const c_void, dst: *mut c_void)

Trait Implementations

impl Default for CPUContext

fn default() -> CPUContext

Returns the “default value” for a type.

impl From<&Device> for CPUContext

fn from(x: &Device) -> CPUContext

Converts to this type from the input type.

impl From<&DeviceOption> for CPUContext

fn from(x: &DeviceOption) -> CPUContext

Converts to this type from the input type.