Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 13 for get_concrete_function (0.28 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top