CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes | Friends
TritonData< IO >::TritonDataEntry Class Reference

Public Member Functions

 TritonDataEntry (const ShapeType &dims, bool noOuterDim, const std::string &name, const std::string &dname)
 
 TritonDataEntry ()
 

Private Member Functions

void computeSizes (int64_t shapeSize, int64_t byteSize, int64_t batchSize)
 
void createObject (IO **ioptr, const std::string &name, const std::string &dname)
 

Private Attributes

size_t byteSizePerBatch_
 
std::shared_ptr< IO > data_
 
ShapeType fullShape_
 
unsigned offset_
 
const uint8_t * output_
 
std::shared_ptr< Resultresult_
 
ShapeView shape_
 
size_t sizeShape_
 
size_t totalByteSize_
 

Friends

class TritonClient
 
class TritonCpuShmResource< IO >
 
class TritonData< IO >
 
class TritonHeapResource< IO >
 
class TritonMemResource< IO >
 

Detailed Description

template<typename IO>
class TritonData< IO >::TritonDataEntry

Definition at line 94 of file TritonData.h.

Constructor & Destructor Documentation

◆ TritonDataEntry() [1/2]

template<typename IO>
TritonData< IO >::TritonDataEntry::TritonDataEntry ( const ShapeType dims,
bool  noOuterDim,
const std::string &  name,
const std::string &  dname 
)
inline

Definition at line 97 of file TritonData.h.

References TritonData< IO >::TritonDataEntry::createObject(), TritonData< IO >::TritonDataEntry::data_, TritonData< IO >::dname(), and Skims_PA_cff::name.

98  : fullShape_(dims),
99  shape_(fullShape_.begin() + (noOuterDim ? 0 : 1), fullShape_.end()),
100  sizeShape_(0),
102  totalByteSize_(0),
103  offset_(0),
104  output_(nullptr) {
105  //create input or output object
106  IO* iotmp;
107  createObject(&iotmp, name, dname);
108  data_.reset(iotmp);
109  }
const uint8_t * output_
Definition: TritonData.h:140
void createObject(IO **ioptr, const std::string &name, const std::string &dname)
std::shared_ptr< IO > data_
Definition: TritonData.h:137
const std::string & dname() const
Definition: TritonData.h:73

◆ TritonDataEntry() [2/2]

template<typename IO>
TritonData< IO >::TritonDataEntry::TritonDataEntry ( )
inline

Definition at line 111 of file TritonData.h.

112  : shape_(fullShape_.begin(), fullShape_.end()),
113  sizeShape_(0),
115  totalByteSize_(0),
116  offset_(0),
117  output_(nullptr) {}
const uint8_t * output_
Definition: TritonData.h:140

Member Function Documentation

◆ computeSizes()

template<typename IO >
void TritonData< IO >::TritonDataEntry::computeSizes ( int64_t  shapeSize,
int64_t  byteSize,
int64_t  batchSize 
)
private

Definition at line 121 of file TritonData.cc.

References TritonData< IO >::byteSize(), and TritonData< IO >::totalByteSize_.

121  {
122  sizeShape_ = shapeSize;
124  totalByteSize_ = byteSizePerBatch_ * batchSize;
125 }
int64_t byteSize() const
Definition: TritonData.h:72

◆ createObject()

template<typename IO>
void TritonData< IO >::TritonDataEntry::createObject ( IO **  ioptr,
const std::string &  name,
const std::string &  dname 
)
private

Friends And Related Function Documentation

◆ TritonClient

template<typename IO>
friend class TritonClient
friend

Definition at line 121 of file TritonData.h.

◆ TritonCpuShmResource< IO >

template<typename IO>
friend class TritonCpuShmResource< IO >
friend

Definition at line 124 of file TritonData.h.

◆ TritonData< IO >

template<typename IO>
friend class TritonData< IO >
friend

Definition at line 120 of file TritonData.h.

◆ TritonHeapResource< IO >

template<typename IO>
friend class TritonHeapResource< IO >
friend

Definition at line 123 of file TritonData.h.

◆ TritonMemResource< IO >

template<typename IO>
friend class TritonMemResource< IO >
friend

Definition at line 122 of file TritonData.h.

Member Data Documentation

◆ byteSizePerBatch_

template<typename IO>
size_t TritonData< IO >::TritonDataEntry::byteSizePerBatch_
private

Definition at line 136 of file TritonData.h.

◆ data_

template<typename IO>
std::shared_ptr<IO> TritonData< IO >::TritonDataEntry::data_
private

Definition at line 137 of file TritonData.h.

Referenced by TritonData< IO >::TritonDataEntry::TritonDataEntry().

◆ fullShape_

template<typename IO>
ShapeType TritonData< IO >::TritonDataEntry::fullShape_
private

Definition at line 134 of file TritonData.h.

◆ offset_

template<typename IO>
unsigned TritonData< IO >::TritonDataEntry::offset_
private

Definition at line 139 of file TritonData.h.

◆ output_

template<typename IO>
const uint8_t* TritonData< IO >::TritonDataEntry::output_
private

Definition at line 140 of file TritonData.h.

◆ result_

template<typename IO>
std::shared_ptr<Result> TritonData< IO >::TritonDataEntry::result_
private

Definition at line 138 of file TritonData.h.

◆ shape_

template<typename IO>
ShapeView TritonData< IO >::TritonDataEntry::shape_
private

Definition at line 135 of file TritonData.h.

◆ sizeShape_

template<typename IO>
size_t TritonData< IO >::TritonDataEntry::sizeShape_
private

Definition at line 136 of file TritonData.h.

◆ totalByteSize_

template<typename IO>
size_t TritonData< IO >::TritonDataEntry::totalByteSize_
private

Definition at line 136 of file TritonData.h.