Internal change

PiperOrigin-RevId: 301434099

official/benchmark/tfhub_memory_usage_benchmark.py

@@ -41,8 +41,14 @@ class TfHubMemoryUsageBenchmark(PerfZeroBenchmark):
         output_dir=output_dir, default_flags=default_flags, **kwargs)
     if hub_model_handle_list:
       for hub_model_handle in hub_model_handle_list.split(';'):
+        # Converts a model handle of the form
+        # https://tfhub.dev/google/nnlm-en-dim128/1 to valid python method name
+        # like google_nnlm_en_dim128_1.
+        hub_model_method_name = hub_model_handle.replace(
+            'https://tfhub.dev',
+            '').replace('/', '_').replace('-', '_').strip('_')
         setattr(
-            self, 'benchmark_' + hub_model_handle,
+            self, 'benchmark_' + hub_model_method_name,
             functools.partial(self.benchmark_memory_usage, hub_model_handle))
 
   def benchmark_memory_usage(