- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for as_graph_def (0.14 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py
"""Gets the number of XlaCallModule ops in the output saved model.""" root = load.load(output_saved_model_path) tf_graph_def = root.signatures['serving_default'].graph.as_graph_def() count = 0 for node_def in tf_graph_def.node: if node_def.op == 'XlaCallModule': count += 1 for function in tf_graph_def.library.function: for node_def in function.node_def: if node_def.op == 'XlaCallModule':
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/tensorflow/python/quantize_model.cc
*module_ref, context.get(), quantization_options, *function_aliases, calibration_data_dir)); // Save and run the calibration model. if (calibration_exported_model.has_graph_def()) { TF_ASSIGN_OR_RETURN(std::string calibration_saved_model_dir, CreateTmpDir()); py_function_library.SaveExportedModel( calibration_saved_model_dir, calibration_exported_model,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 23.8K bytes - Viewed (0)