Home
last modified time | relevance | path

Searched defs:weight_scales (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h410 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
420 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h414 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
424 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils.cc267 std::vector<float> weight_scales(channel_dim_size); in AdjustWeightsForBiasScale() local
671 const float* weight_scales, in SymmetricPerChannelBiasQuantize()
Dquantization_utils_test.cc855 const std::vector<float> weight_scales = {0.5, 1.0}; in TEST_F() local
Dquantize_model.cc304 std::vector<float> weight_scales = weight_tensor->quantization->scale; in QuantizeBias() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream.cc555 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized()
575 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized()