Searched refs:iweights (Results 1 – 1 of 1) sorted by relevance
960 void ReorderCKtoKC(const TRT_ShapedWeights& iweights, in ReorderCKtoKC() argument962 const int c = iweights.shape_.d[0]; in ReorderCKtoKC()963 const int k = iweights.shape_.d[1]; in ReorderCKtoKC()968 switch (iweights.TrtDType()) { in ReorderCKtoKC()970 Reorder2({k, c}, static_cast<float const*>(iweights.GetValues()), in ReorderCKtoKC()975 Reorder2({k, c}, static_cast<Eigen::half const*>(iweights.GetValues()), in ReorderCKtoKC()982 << DebugString(iweights.TrtDType()); in ReorderCKtoKC()986 void ReorderRSCKToKCRS(const TRT_ShapedWeights& iweights, in ReorderRSCKToKCRS() argument988 CHECK(iweights.TrtDType() == oweights->TrtDType()); in ReorderRSCKToKCRS()989 CHECK_EQ(iweights.size_bytes(), oweights->size_bytes()); in ReorderRSCKToKCRS()[all …]