- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for as_graph_def (0.25 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) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
@tf.function def foo(x, y): return mlir_passthrough_op([x, y], mlir_module, Toutputs=[tf.float32]) graph_def = foo.get_concrete_function(tf.TensorSpec([10], tf.float32), tf.TensorSpec([10], tf.float32)).graph.as_graph_def() ``` }]; let arguments = (ins Variadic<TF_Tensor>:$inputs, StrAttr:$mlir_module ); let results = (outs Variadic<TF_Tensor>:$outputs
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)