- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GatherModel (0.08 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py
def _create_gather_model(self, input_type, use_variable): class GatherModel(autotrackable.AutoTrackable): """A simple model with a single gather.""" def __init__(self, use_variable): """Initializes a GatherModel. Args: use_variable: If True, creates a variable for weight. """ super(GatherModel, self).__init__() w_val = np.random.randn(128, 32).astype('f4')
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 08:51:46 UTC 2024 - 51.2K bytes - Viewed (0)