- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for get_concrete_function (0.28 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py
return {'output': out} model = MatmulModel(weight_shape) saved_model_save.save( model, saved_model_path, signatures=model.matmul.get_concrete_function( tensor_spec.TensorSpec( shape=input_shape, dtype=dtypes.float32, name='input_tensor' ) ), ) return model def _any_log_contains(
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/c/experimental/saved_model/core/concrete_function.h
#include "tensorflow/c/experimental/saved_model/core/function_metadata.h" namespace tensorflow { // ConcreteFunctions correspond to an instance of a tf.function with a known set // of inputs (either through get_concrete_function) or an input_signature. // ConcreteFunction attempts to preserve the user-facing semantics of the // tf.function python API and can take a limited set of types as arguments
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 29 15:50:58 UTC 2021 - 2.4K bytes - Viewed (0) -
tensorflow/cc/saved_model/testdata/generate_chunked_models.py
root = module.Module() root.c = constant_op.constant(np.random.random_sample([150, 150])) constants.debug_set_max_size(80000) root.get_c = def_function.function(lambda: root.c) signatures = root.get_c.get_concrete_function() save.save(root, non_chunked_dir, signatures=signatures, options=save_options.SaveOptions(experimental_image_format=False)) def generate_chunked_model(non_chunked_dir: str, chunked_dir: str):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 21:43:11 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/debuginfo/concrete_function_error.py
@tf.function( input_signature=[tf.TensorSpec(shape=[3, 3], dtype=tf.float32)]) def model(x): y = tf.math.betainc(x, 0.5, 1.0) # Not supported return y + y func = model.get_concrete_function() converter = tf.lite.TFLiteConverter.from_concrete_functions([func], model) converter.convert() # pylint: disable=line-too-long # CHECK-LABEL: testConcreteFunctionDebugInfo
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/concurrency_test.py
), } root = ModelWithAdd() temp_path = self.create_tempdir().full_path saved_model_save.save( root, temp_path, signatures=root.add.get_concrete_function() ) quantization_options = quant_opts_pb2.QuantizationOptions( quantization_method=quant_opts_pb2.QuantizationMethod(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Sep 11 00:47:05 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
# Create and save a model having 2 signatures. model = MultipleSignatureModel() signatures = { 'sig1': model.matmul.get_concrete_function( tensor_spec.TensorSpec(shape=(1, 4), dtype=dtypes.float32) ), 'sig2': model.conv.get_concrete_function( tensor_spec.TensorSpec(shape=(1, 3, 4, 3), dtype=dtypes.float32) ), } saved_model_save.save(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py
return {'output': out} model = TwoMatmulModel() input_shape = (1, 2) save.save( model, self._input_saved_model_path, signatures=model.matmul.get_concrete_function( tensor_spec.TensorSpec( shape=input_shape, dtype=dtypes.float32, name='input_tensor' ) ), ) def data_gen() -> repr_dataset.RepresentativeDataset:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py
model = MatmulModel(weight_shape, bias_size, activation_fn) saved_model_save.save( model, saved_model_path, signatures=model.matmul.get_concrete_function( tensor_spec.TensorSpec( shape=input_shape, dtype=dtypes.float32, name='input_tensor' ) ), ) return model
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 08:51:46 UTC 2024 - 51.2K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/function.h
private: struct ConcreteFunction { tensorflow::AbstractFunctionPtr trace; TaggedValue input_signature; TaggedValue output_signature; }; tensorflow::StatusOr<ConcreteFunction> GetConcreteFunction(TaggedValue) const; std::vector<ConcreteFunction> concrete_fns_; }; } // namespace libtf } // namespace tf
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 30 21:44:45 UTC 2022 - 1.9K bytes - Viewed (0) -
tensorflow/cc/saved_model/experimental/public/saved_model_api.h
// If status is not OK, returns nullptr. Otherwise, returns a // tensorflow::cc::ConcreteFunction pointer. The lifetime of this pointer // is bound to SavedModelAPI it was loaded from. ConcreteFunction* GetConcreteFunction(const std::string& function_path, Status* status); // Retrieve a function from the TF SavedModel via a SignatureDef key. // // Params:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 04 00:45:47 UTC 2020 - 6.4K bytes - Viewed (0)