- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for tensorspecs (0.38 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py
input_signature=[ tensor_spec.TensorSpec( name='x', shape=x_signature, dtype=dtypes.float32 ) ] ) def einsum_with_kernel(self, x: core.Tensor) -> Mapping[str, core.Tensor]: return self._einsum(x, self._kernel) @def_function.function( input_signature=[ tensor_spec.TensorSpec(
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/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py
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: for _ in range(50):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0)