Searched refs:iweights (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 797 void ReorderCKtoKC(const TRT_ShapedWeights& iweights, in ReorderCKtoKC() argument
799 const int c = iweights.shape_.d[0]; in ReorderCKtoKC()
800 const int k = iweights.shape_.d[1]; in ReorderCKtoKC()
805 switch (iweights.type_) { in ReorderCKtoKC()
807 Reorder2({k, c}, static_cast<float const*>(iweights.GetValues()), in ReorderCKtoKC()
816 {k, c}, static_cast<Eigen::half const*>(iweights.GetValues()), in ReorderCKtoKC()
824 << DataTypeString(iweights.type_); in ReorderCKtoKC()
828 void ReorderRSCKToKCRS(const TRT_ShapedWeights& iweights, in ReorderRSCKToKCRS() argument
830 CHECK_EQ(iweights.type_, oweights->type_); in ReorderRSCKToKCRS()
831 CHECK_EQ(iweights.size_bytes(), oweights->size_bytes()); in ReorderRSCKToKCRS()
[all …]
|