Searched refs:num_scales (Results 1 – 2 of 2) sorted by relevance
381 const size_t num_scales = src_quantization->scale()->size(); in ParseQuantization() local395 if (num_scales != 1 && in ParseQuantization()397 num_scales != dims[src_quantization->quantized_dimension()])) { in ParseQuantization()401 dims[src_quantization->quantized_dimension()], num_scales); in ParseQuantization()409 affine_quantization->scale = TfLiteFloatArrayCreate(num_scales); in ParseQuantization()410 affine_quantization->zero_point = TfLiteIntArrayCreate(num_scales); in ParseQuantization()411 for (size_t i = 0; i < num_scales; ++i) { in ParseQuantization()
220 const size_t num_scales = params->scale->size; in ExportTensors() local223 params->scale->data + num_scales); in ExportTensors()225 params->zero_point->data, params->zero_point->data + num_scales); in ExportTensors()