nnkit::support::caffe::OutputBlobContext< DType > Class Template Reference

#include <OutputBlobContext.h>

Collaboration diagram for nnkit::support::caffe::OutputBlobContext< DType >:

Public Member Functions
	OutputBlobContext (::caffe::Net< DType > &net)
uint32_t	size (void) const override
std::string	name (uint32_t n) const override
::caffe::Blob< DType > *	blob (uint32_t n) override
Public Member Functions inherited from nnkit::support::caffe::BlobContext< DType >
virtual	~BlobContext ()=default
DType *	region (uint32_t n)

Detailed Description

template<typename DType>
class nnkit::support::caffe::OutputBlobContext< DType >

Definition at line 31 of file OutputBlobContext.h.

Constructor & Destructor Documentation

OutputBlobContext()

template<typename DType >

nnkit::support::caffe::OutputBlobContext< DType >::OutputBlobContext ( ::caffe::Net< DType > &  net )

inline

Definition at line 34 of file OutputBlobContext.h.

                                          : _net(net)
  {
    // DO NOTHING
  }

Member Function Documentation

blob()

template<typename DType >

::caffe::Blob< DType > * nnkit::support::caffe::OutputBlobContext< DType >::blob ( uint32_t  n )

inlineoverridevirtual

Implements nnkit::support::caffe::BlobContext< DType >.

Definition at line 47 of file OutputBlobContext.h.

{ return _net.output_blobs().at(n); }

name()

template<typename DType >

std::string nnkit::support::caffe::OutputBlobContext< DType >::name ( uint32_t  n ) const

inlineoverridevirtual

Implements nnkit::support::caffe::BlobContext< DType >.

Definition at line 42 of file OutputBlobContext.h.

  {
    return _net.blob_names().at(_net.output_blob_indices().at(n));
  }

size()

template<typename DType >

uint32_t nnkit::support::caffe::OutputBlobContext< DType >::size ( void  ) const

inlineoverridevirtual

Implements nnkit::support::caffe::BlobContext< DType >.

Definition at line 40 of file OutputBlobContext.h.

{ return _net.num_outputs(); }

---

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

• compiler/nnkit-caffe/support/include/nnkit/support/caffe/OutputBlobContext.h