Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::detail::WorkspaceStack Class Reference

Public Member Functions

std::shared_ptr< WorkspacepushForwardWorkspace (Workspace *parent_ws, const std::unordered_map< std::string, std::string > &blob_bindings)
 
std::shared_ptr< WorkspacepopGradientWorkspace (Workspace *parent_ws, const std::unordered_map< std::string, std::string > &grad_blob_bindings)
 
std::shared_ptr< WorkspacereuseLastForwardWorkspace (Workspace *parent_ws, const std::unordered_map< std::string, std::string > &blob_bindings)
 
void clear ()
 
bool empty () const
 

Detailed Description

Definition at line 39 of file create_scope_op.h.


The documentation for this class was generated from the following file: