Searched refs:oweights (Results 1 – 1 of 1) sorted by relevance
961 TRT_ShapedWeights* oweights) { in ReorderCKtoKC() argument964 oweights->shape_.d[0] = k; in ReorderCKtoKC()965 oweights->shape_.d[1] = c; in ReorderCKtoKC()971 istrides, static_cast<float*>(oweights->GetValues()), ostrides); in ReorderCKtoKC()976 istrides, static_cast<Eigen::half*>(oweights->GetValues()), in ReorderCKtoKC()987 TRT_ShapedWeights* oweights, const int num_groups) { in ReorderRSCKToKCRS() argument988 CHECK(iweights.TrtDType() == oweights->TrtDType()); in ReorderRSCKToKCRS()989 CHECK_EQ(iweights.size_bytes(), oweights->size_bytes()); in ReorderRSCKToKCRS()1001 oweights->shape_.d[0] = k / num_groups; in ReorderRSCKToKCRS()1002 oweights->shape_.d[1] = c * num_groups; in ReorderRSCKToKCRS()[all …]