Searched defs:weight_scales (Results 1 – 4 of 4) sorted by relevance
398 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()408 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
389 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()399 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
155 const std::vector<float>& weight_scales = weight_tensor->quantization->scale; in SymmetricPerChannelBiasQuantize() local
1160 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized()1182 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized()