Searched refs:weights_drsck (Results 1 – 1 of 1) sorted by relevance
3247 const TRT_ShapedWeights weights_drsck = inputs.at(1).weights(); in ConvertConv3DHelper() local3248 if (weights_drsck.shape_.nbDims != kNumDims) { in ConvertConv3DHelper()3301 params->weight_store->GetTempWeights(weights_drsck); in ConvertConv3DHelper()3350 params->weight_store->GetTempWeights(weights_drsck); in ConvertConv3DHelper()3351 ReorderDRSCKToKCDRS(weights_drsck, &weights, num_groups); in ConvertConv3DHelper()