Home
last modified time | relevance | path

Searched refs:TrtDType (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc968 switch (iweights.TrtDType()) { in ReorderCKtoKC()
982 << DebugString(iweights.TrtDType()); in ReorderCKtoKC()
988 CHECK(iweights.TrtDType() == oweights->TrtDType()); in ReorderRSCKToKCRS()
1007 switch (iweights.TrtDType()) { in ReorderRSCKToKCRS()
1022 << DebugString(iweights.TrtDType()); in ReorderRSCKToKCRS()
1037 DCHECK(iweights.TrtDType() == oweights->TrtDType()); in ReorderDRSCKToKCDRS()
1069 switch (iweights.TrtDType()) { in ReorderDRSCKToKCDRS()
1083 << DebugString(iweights.TrtDType()); in ReorderDRSCKToKCDRS()
1631 switch (weights.TrtDType()) { in GetWeightRange()
1656 DebugString(weights.TrtDType())); in GetWeightRange()
[all …]
Dconvert_nodes.h212 nvinfer1::DataType TrtDType() const { return type_; } in TrtDType() function
250 return GetTempWeights(weights.TrtDType(), weights.shape_); in GetTempWeights()
Dconvert_nodes_test.cc288 lhs.TrtDType() == rhs.TrtDType() && lhs.GetValues() == rhs.GetValues(); in TrtShapedWeightsEquals()