Searched refs:weight_t (Results 1 – 1 of 1) sorted by relevance
431 const Tensor& weight_t = *weight_oi_.AccessTensor(context); in CacheWeight() local434 if (weight_t.NumElements() > 0) { in CacheWeight()475 const Tensor& weight_t = *weight_oi_.AccessTensor(context); in GetCachedWeight() local485 const_cast<Tweight*>(weight_t.flat<Tweight>().data())); in GetCachedWeight()