- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _get_function_aliases (0.17 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py
count += 1 for function in tf_graph_def.library.function: for node_def in function.node_def: if node_def.op == 'XlaCallModule': count += 1 return count def _get_function_aliases( self, output_saved_model_path: str, tags: List[str] ) -> dict[str, str]: """Gets the function aliases in the output saved model.""" loader = loader_impl.SavedModelLoader(output_saved_model_path)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 18.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py
self.assertNotRegex(module_str, r'stablehlo.clamp.*\n.*return') self.assertNotRegex(module_str, r'stablehlo.maximum.*\n.*return') if has_func_alias: func_aliases = self._get_function_aliases( self._output_saved_model_path, [tag_constants.SERVING] ) self.assertCountEqual( func_aliases.values(), [quantize_model_test_base.FUNC_ALIAS] )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0)