DynamicGraph_AnyTensorGroup

public protocol DynamicGraph_AnyTensorGroup : DynamicGraph_Any

This protocol defines a generic constructs such that we can use DynamicGraph.AnyTensorGroup to represent either a collection of tensors from multiple GPUs or one tensor. It has a typed version DynamicGraph.TensorGroup to enforce type constraint.


                    
                    
                    AnyTensor

Declaration

Swift

associatedtype AnyTensor : DynamicGraph_Any


                    
                    
                    exec(cmd:hint:inputs:outputSize:streamContext:)

Declaration

Swift

static func exec(
  cmd: ccv_nnc_cmd_t, hint: ccv_nnc_hint_t, inputs: [DynamicGraph_Any?], outputSize: Int32,
  streamContext: StreamContext?
) -> [AnyTensor]


                    
                    
                    exec(cmd:hint:inputs:outputs:streamContext:)

Declaration

Swift

static func exec(
  cmd: ccv_nnc_cmd_t, hint: ccv_nnc_hint_t, inputs: [DynamicGraph_Any?],
  outputs: [DynamicGraph_Any],
  streamContext: StreamContext?)


                    
                    
                    evaluate(model:isTest:dataParallel:inputs:outputSize:streamContext:)

Declaration

Swift

static func evaluate(
  model: OpaquePointer, isTest: Bool, dataParallel: inout Int?, inputs: [DynamicGraph_Any?],
  outputSize: Int32, streamContext: StreamContext?
) -> [AnyTensor]