Searched refs:TRT_ShapedWeights (Results 1 – 3 of 3) sorted by relevance
174 class TRT_ShapedWeights {176 explicit TRT_ShapedWeights(DataType type = DT_FLOAT);182 TRT_ShapedWeights(const TRT_ShapedWeights& rhs);216 TRT_ShapedWeights(DataType type, nvinfer1::Dims dims, Tensor tensor);237 TRT_ShapedWeights GetTempWeights(DataType type, const nvinfer1::Dims& dims);241 TRT_ShapedWeights GetTempWeights(const TRT_ShapedWeights& weights) { in GetTempWeights()273 explicit TRT_TensorOrWeights(const TRT_ShapedWeights& weights);286 TRT_ShapedWeights& weights() { in weights()291 const TRT_ShapedWeights& weights() const { in weights()337 TRT_ShapedWeights weights_;[all …]
446 const TRT_ShapedWeights& weights, const nvinfer1::Dims& dims) { in CreateConstantLayer()479 TRT_ShapedWeights weights = in CreateBroadcastableScalarConstant()611 TRT_ShapedWeights::TRT_ShapedWeights(DataType type) : type_(type) { in TRT_ShapedWeights() function in tensorflow::tensorrt::convert::TRT_ShapedWeights615 TRT_ShapedWeights::TRT_ShapedWeights(DataType type, nvinfer1::Dims dims, in TRT_ShapedWeights() function in tensorflow::tensorrt::convert::TRT_ShapedWeights619 TRT_ShapedWeights::TRT_ShapedWeights(const TRT_ShapedWeights& rhs) in TRT_ShapedWeights() function in tensorflow::tensorrt::convert::TRT_ShapedWeights622 int64_t TRT_ShapedWeights::count() const { return TrtDimsNumElements(shape_); } in count()624 nvinfer1::Weights TRT_ShapedWeights::GetTrtWeights() const { in GetTrtWeights()630 size_t TRT_ShapedWeights::size_bytes() const { in size_bytes()634 string TRT_ShapedWeights::DebugString() const { in DebugString()715 TRT_TensorOrWeights::TRT_TensorOrWeights(const TRT_ShapedWeights& weights) in TRT_TensorOrWeights()[all …]
187 bool TrtShapedWeightsEquals(const TRT_ShapedWeights& lhs, in TrtShapedWeightsEquals()188 const TRT_ShapedWeights& rhs) { in TrtShapedWeightsEquals()194 void ValidateWeights(const TRT_ShapedWeights& weights, in ValidateWeights()271 TRT_ShapedWeights weights; in TEST()272 TRT_ShapedWeights copy(weights); in TEST()286 TRT_ShapedWeights weights(DT_FLOAT); in TEST()287 TRT_ShapedWeights copy(weights); in TEST()302 TRT_ShapedWeights weights = in TEST()304 TRT_ShapedWeights copy(weights); in TEST()381 TRT_ShapedWeights weights; in TEST()[all …]