Searched refs:weights_rsck (Results 1 – 1 of 1) sorted by relevance
1784 TRT_ShapedWeights weights_rsck = inputs.at(1).weights(); in ConvertConv2DHelper() local1785 if (weights_rsck.shape_.nbDims != 4) { in ConvertConv2DHelper()1841 weights_rsck = ConvertFP32ToFP16(params->weight_store, weights_rsck); in ConvertConv2DHelper()1847 params->weight_store->GetTempWeights(weights_rsck); in ConvertConv2DHelper()1848 ReorderRSCKToKCRS(weights_rsck, &weights, num_groups); in ConvertConv2DHelper()