Searched refs:weights_rsck (Results 1 – 1 of 1) sorted by relevance
2292 TRT_ShapedWeights weights_rsck = inputs.at(1).weights(); in ConvertConv2DHelper() local2293 if (weights_rsck.shape_.nbDims != 4) { in ConvertConv2DHelper()2359 params->weight_store->GetTempWeights(weights_rsck); in ConvertConv2DHelper()2360 ReorderRSCKToKCRS(weights_rsck, &weights, num_groups); in ConvertConv2DHelper()