Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for get_concrete_function (0.27 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/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)
  3. 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)
  4. RELEASE.md

            sometimes incorrectly ignored.
        *   functions returned by the `get_concrete_function` method of
            `tf.function` objects can now be called with arguments consistent with
            the original arguments or type specs passed to `get_concrete_function`.
            This calling convention is now the preferred way to use concrete
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 730.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

       return %ret : tensor<10x10xf32>
    }
    '''
    
    @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
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top