Searched refs:oweights (Results 1 – 1 of 1) sorted by relevance
798 TRT_ShapedWeights* oweights) { in ReorderCKtoKC() argument801 oweights->shape_.d[0] = k; in ReorderCKtoKC()802 oweights->shape_.d[1] = c; in ReorderCKtoKC()810 static_cast<float*>(const_cast<void*>(oweights->GetValues())), in ReorderCKtoKC()818 static_cast<Eigen::half*>(const_cast<void*>(oweights->GetValues())), in ReorderCKtoKC()829 TRT_ShapedWeights* oweights, const int num_groups) { in ReorderRSCKToKCRS() argument830 CHECK_EQ(iweights.type_, oweights->type_); in ReorderRSCKToKCRS()831 CHECK_EQ(iweights.size_bytes(), oweights->size_bytes()); in ReorderRSCKToKCRS()843 oweights->shape_.d[0] = k / num_groups; in ReorderRSCKToKCRS()844 oweights->shape_.d[1] = c * num_groups; in ReorderRSCKToKCRS()[all …]