- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for TwoMatmulModel (0.3 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py
rng = np.random.default_rng(1230) random_tensor_gen_fn = lambda shape: rng.uniform( low=-1.0, high=1.0, size=shape ).astype(np.float32) class TwoMatmulModel(module.Module): """A model with two matmul ops.""" @def_function.function def matmul(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]: """Performs a matrix multiplication.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0)