Searched defs:weight_scales (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 417 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
427 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 441 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
451 const DeviceMemory<float>& weight_scales, in DoMatMulQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | testing_helpers_test.cc | 45 float weight_scales[] = {0.5, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
|
| D | test_helpers.cc | 1020 float input_scale, float* weight_scales, float* scales, int* zero_points, in CreatePerChannelQuantizedBiasTensor()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 257 std::vector<float> weight_scales(channel_dim_size); in AdjustWeightsForBiasScale() local
658 const float* weight_scales, in SymmetricPerChannelBiasQuantize()
|
| D | quantization_utils_test.cc | 786 const std::vector<float> weight_scales = {0.5, 1.0}; in TEST_F() local
|
| D | quantize_model.cc | 153 std::vector<float> weight_scales = weight_tensor->quantization->scale; in QuantizeBias() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 765 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized()
785 const DeviceMemory<float> &weight_scales, in ThenMatMulQuantized()
|