ONE - On-device Neural Engine
Loading...
Searching...
No Matches
nnkit::support::caffe::InputBlobContext< DType > Class Template Referencefinal

#include <InputBlobContext.h>

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

Public Member Functions

 InputBlobContext (::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::InputBlobContext< DType >

Definition at line 31 of file InputBlobContext.h.

Constructor & Destructor Documentation

◆ InputBlobContext()

template<typename DType >
nnkit::support::caffe::InputBlobContext< DType >::InputBlobContext ( ::caffe::Net< DType > &  net)
inline

Definition at line 34 of file InputBlobContext.h.

34 : _net(net)
35 {
36 // DO NOTHING
37 }

Member Function Documentation

◆ blob()

template<typename DType >
::caffe::Blob< DType > * nnkit::support::caffe::InputBlobContext< DType >::blob ( uint32_t  n)
inlineoverridevirtual

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

Definition at line 47 of file InputBlobContext.h.

47{ return _net.input_blobs().at(n); }

◆ name()

template<typename DType >
std::string nnkit::support::caffe::InputBlobContext< DType >::name ( uint32_t  n) const
inlineoverridevirtual

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

Definition at line 42 of file InputBlobContext.h.

43 {
44 return _net.blob_names().at(_net.input_blob_indices().at(n));
45 }

◆ size()

template<typename DType >
uint32_t nnkit::support::caffe::InputBlobContext< DType >::size ( void  ) const
inlineoverridevirtual

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

Definition at line 40 of file InputBlobContext.h.

40{ return _net.num_inputs(); }

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