Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for assertCountEqual (0.29 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

        converted_model = quantize_model.quantize(
            self._input_saved_model_path,
            representative_dataset=self._simple_model_data_gen(),
        )
    
        self.assertIsNotNone(converted_model)
        self.assertCountEqual(
            converted_model.signatures._signatures.keys(), {'serving_default'}
        )
    
        # Indirectly prove that it is performing a static-range quantization
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py

        root = load.load(self._output_saved_model_path)
        self.assertCountEqual(root.signatures.keys(), {'serving_default'})
        new_outputs_1 = root.signatures['serving_default'](
            x=ops.convert_to_tensor(input_data)
        )
    
        root = load.load(output_saved_model_path_2)
        self.assertCountEqual(root.signatures.keys(), {'serving_default'})
        new_outputs_2 = root.signatures['serving_default'](
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 51.4K bytes
    - Viewed (0)
Back to top