Function quantizationTest

tfjs-core/src/io/weights_loader_test.ts:442–482

(quantizationDtype: 'uint8'|'uint16')

  });

  const quantizationTest = async (quantizationDtype: 'uint8'|'uint16') => {
    const arrayType = quantizationDtype === 'uint8' ? Uint8Array : Uint16Array;
    setupFakeWeightFiles(
        {'./weightfile0': new arrayType([0, 48, 255, 0, 48, 255])});

    const manifest: WeightsManifestConfig = [{
      'paths': ['weightfile0'],
      'weights': [
        {
          'name': 'weight0',
          'dtype': 'float32',
          'shape': [3],
          'quantization': {'min': -1, 'scale': 0.1, 'dtype': quantizationDtype}
        },
        {
          'name': 'weight1',
          'dtype': 'int32',
          'shape': [3],
          'quantization': {'min': -1, 'scale': 0.1, 'dtype': quantizationDtype}
        }
      ]
    }];

    const weightsNamesToFetch = ['weight0', 'weight1'];
    const weights =
        await tf.io.loadWeights(manifest, './', weightsNamesToFetch);
    expect((tf.env().platform.fetch as jasmine.Spy).calls.count()).toBe(1);

    const weightNames = Object.keys(weights);
    expect(weightNames.length).toEqual(weightsNamesToFetch.length);

    const weight0 = weights['weight0'];
    expectArraysClose(await weight0.data(), [-1, 3.8, 24.5]);
    expect(weight0.shape).toEqual([3]);
    expect(weight0.dtype).toEqual('float32');

    const weight1 = weights['weight1'];
    expectArraysEqual(await weight1.data(), [-1, 4, 25]);
    expect(weight1.shape).toEqual([3]);
    expect(weight1.dtype).toEqual('int32');
  };

  it('quantized weights (uint8)', async () => {
    await quantizationTest('uint8');