Searched refs:CacheWeight (Results 1 – 3 of 3) sorted by relevance
231 this->CacheWeight(ctx, matmul_pd, cached_weight_data, weight_tensor, in Compute()
270 this->CacheWeight(context, matmul_fwd_pd, weight_data, in Compute()
423 void CacheWeight( in CacheWeight() function