ForwardGraph

public class ForwardGraph: ComputationGraph

ForwardGraph is a subclass of ComputationGraph that can only do forward computation. Thus, it has mooptimization during forward computation and the internal results are not accessable.

allocateAllTensors()
When allocate tensors for graph, this function will try best to reuse tensors for TensorSymbol whose SymbolDataSource is Calculation
Declaration
Swift

override public func allocateAllTensors()
Show on GitHub
checkShapeChain()
Since calcualtion tensor symbols shares tensor. We need to correct its shape before verify.
Declaration
Swift

override public func checkShapeChain() -> (valid: Bool, msg: String)
Show on GitHub


                    
                    
                    stageOrderCalculate(mode:)

Since calcualtion tensor symbols shares tensor. We need to assign its shape before calcultaion.

Declaration

Swift

override internal func stageOrderCalculate(mode: OperatorComputationMode)

Parameters


                                mode

mode

Show on GitHub