pub struct OperationGraphSpec {
pub tensors: Vec<TensorSpec>,
pub ops: Vec<OpSpec>,
pub name: String,
}Expand description
Top-level operation graph.
Fields§
§tensors: Vec<TensorSpec>§ops: Vec<OpSpec>§name: StringOptional name for diagnostics.
Implementations§
Source§impl OperationGraphSpec
impl OperationGraphSpec
pub fn new(name: impl Into<String>) -> Self
pub fn add_tensor(&mut self, t: TensorSpec) -> i64
pub fn add_op(&mut self, op: OpSpec)
Sourcepub fn build_into(
&self,
_handle: cudnnHandle_t,
) -> Result<BackendDescriptor, GpuError>
pub fn build_into( &self, _handle: cudnnHandle_t, ) -> Result<BackendDescriptor, GpuError>
Drive cudnnBackendCreateDescriptor for every tensor and op,
then build an OPERATION_GRAPH_DESCRIPTOR. Returns the
finalised graph descriptor.
On hosts without cuDNN, this short-circuits with
LibraryError("cudnn-frontend graph build path requires a real handle").
Trait Implementations§
Source§impl Clone for OperationGraphSpec
impl Clone for OperationGraphSpec
Source§fn clone(&self) -> OperationGraphSpec
fn clone(&self) -> OperationGraphSpec
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for OperationGraphSpec
impl RefUnwindSafe for OperationGraphSpec
impl Send for OperationGraphSpec
impl Sync for OperationGraphSpec
impl Unpin for OperationGraphSpec
impl UnsafeUnpin for OperationGraphSpec
impl UnwindSafe for OperationGraphSpec
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more