Searched refs:TRT_ShapedWeights (Results 1 – 3 of 3) sorted by relevance
170 class TRT_ShapedWeights {172 explicit TRT_ShapedWeights(179 TRT_ShapedWeights(const TRT_ShapedWeights& rhs);220 TRT_ShapedWeights(nvinfer1::DataType type, nvinfer1::Dims dims,244 TRT_ShapedWeights GetTempWeights(nvinfer1::DataType trt_type,249 TRT_ShapedWeights GetTempWeights(const TRT_ShapedWeights& weights) { in GetTempWeights()281 explicit TRT_TensorOrWeights(const TRT_ShapedWeights& weights);292 TRT_ShapedWeights& weights() { in weights()297 const TRT_ShapedWeights& weights() const { in weights()348 TRT_ShapedWeights weights_;[all …]
471 const TRT_ShapedWeights& weights, const nvinfer1::Dims& dims) { in CreateConstantLayer()495 TRT_ShapedWeights weights =690 TRT_ShapedWeights::TRT_ShapedWeights(nvinfer1::DataType type) : type_(type) { in TRT_ShapedWeights() function in tensorflow::tensorrt::convert::TRT_ShapedWeights694 TRT_ShapedWeights::TRT_ShapedWeights(nvinfer1::DataType type, in TRT_ShapedWeights() function in tensorflow::tensorrt::convert::TRT_ShapedWeights698 TRT_ShapedWeights::TRT_ShapedWeights(const TRT_ShapedWeights& rhs) in TRT_ShapedWeights() function in tensorflow::tensorrt::convert::TRT_ShapedWeights701 int64_t TRT_ShapedWeights::count() const { in count()705 nvinfer1::Weights TRT_ShapedWeights::GetTrtWeights() const { in GetTrtWeights()710 Status TRT_ShapedWeights::SetValues(T value) { in SetValues()734 size_t TRT_ShapedWeights::size_bytes() const { in size_bytes()754 string TRT_ShapedWeights::DebugString() const { in DebugString()[all …]
285 bool TrtShapedWeightsEquals(const TRT_ShapedWeights& lhs, in TrtShapedWeightsEquals()286 const TRT_ShapedWeights& rhs) { in TrtShapedWeightsEquals()292 void ValidateWeights(const TRT_ShapedWeights& weights, in ValidateWeights()403 TRT_ShapedWeights weights; in TEST()404 TRT_ShapedWeights copy(weights); in TEST()418 TRT_ShapedWeights weights(nvinfer1::DataType::kFLOAT); in TEST()419 TRT_ShapedWeights copy(weights); in TEST()434 TRT_ShapedWeights weights = in TEST()436 TRT_ShapedWeights copy(weights); in TEST()513 TRT_ShapedWeights weights; in TEST()[all …]