allocateWeights(int64_t elems, std::uniform_real_distribution< float > distribution=std::uniform_real_distribution< float >(-0.01f, 0.01F)) | nvcaffeparser1::CaffeWeightFactory | |
allocateWeights(int64_t elems, std::normal_distribution< float > distribution) | nvcaffeparser1::CaffeWeightFactory | |
CaffeWeightFactory(const trtcaffe::NetParameter &msg, nvinfer1::DataType dataType, std::vector< void * > &tmpAllocs, bool isInitialized) | nvcaffeparser1::CaffeWeightFactory | |
checkForNans(const void *values, int count, const std::string &layerName) | nvcaffeparser1::CaffeWeightFactory | private |
convert(nvinfer1::Weights &weights, nvinfer1::DataType targetType) | nvcaffeparser1::CaffeWeightFactory | |
convert(nvinfer1::Weights &weights) | nvcaffeparser1::CaffeWeightFactory | |
generator | nvcaffeparser1::CaffeWeightFactory | private |
getAllWeights(const std::string &layerName) | nvcaffeparser1::CaffeWeightFactory | |
getBlob(const std::string &layerName, int index) | nvcaffeparser1::CaffeWeightFactory | |
getBlobProtoData(const trtcaffe::BlobProto &blobMsg, trtcaffe::Type type, std::vector< void * > &tmpAllocs) | nvcaffeparser1::CaffeWeightFactory | static |
getBlobProtoDataType(const trtcaffe::BlobProto &blobMsg) | nvcaffeparser1::CaffeWeightFactory | static |
getBlobsSize(const std::string &layerName) | nvcaffeparser1::CaffeWeightFactory | |
getDataType() const | nvcaffeparser1::CaffeWeightFactory | |
getDataTypeSize() const | nvcaffeparser1::CaffeWeightFactory | |
getNullWeights() | nvcaffeparser1::CaffeWeightFactory | |
getTmpAllocs() | nvcaffeparser1::CaffeWeightFactory | |
getWeights(const trtcaffe::BlobProto &blobMsg, const std::string &layerName) | nvcaffeparser1::CaffeWeightFactory | private |
isInitialized() | nvcaffeparser1::CaffeWeightFactory | |
isOK() | nvcaffeparser1::CaffeWeightFactory | |
mDataType | nvcaffeparser1::CaffeWeightFactory | private |
mInitialized | nvcaffeparser1::CaffeWeightFactory | private |
mMsg | nvcaffeparser1::CaffeWeightFactory | private |
mOK | nvcaffeparser1::CaffeWeightFactory | private |
mRef | nvcaffeparser1::CaffeWeightFactory | private |
mTmpAllocs | nvcaffeparser1::CaffeWeightFactory | private |
operator()(const std::string &layerName, WeightType weightType) | nvcaffeparser1::CaffeWeightFactory | virtual |
sizeOfCaffeType(trtcaffe::Type type) | nvcaffeparser1::CaffeWeightFactory | static |