Searched refs:weights_b (Results 1 – 1 of 1) sorted by relevance
5419 TRT_ShapedWeights weights_b, in ConvertFullyConnectedHelper() argument5431 TRT_ShapedWeights weights(weights_b.TrtDType()); in ConvertFullyConnectedHelper()5433 weights = params->weight_store->GetTempWeights(weights_b); in ConvertFullyConnectedHelper()5434 ReorderCKtoKC(weights_b, &weights); in ConvertFullyConnectedHelper()5436 weights = weights_b; in ConvertFullyConnectedHelper()