- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _create_add_model (0.1 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py
"""Performs a gather operation.""" out = array_ops.gather_v2(self.w, input_tensor) return {'output': out} return GatherModel(use_variable) def _create_add_model( self, shape: Sequence[int], saved_model_path: str, ) -> module.Module: class AddModel(module.Module): """A simple model with a single add.""" def __init__(self):
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/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py
) @test_util.run_in_graph_and_eager_modes def test_add_ptq_model( self, shape_dynamic: bool, ): input_shape = (None, 3, 4, 3) if shape_dynamic else (2, 3, 4, 3) self._create_add_model( input_shape, self._input_saved_model_path, ) # Generate model input data. rng = np.random.default_rng(seed=42)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0)