6biSrSSKrSSKrSSKJr SSKJr SrSr S Sjr "SS \R5r S S jr g) zCommon utils for benchmarks.N) callbacks)distribution_utilcvSU;dSU;a [S5eURS5SRS5$)aLSplit the suffix of the benchmark name. For example, for the name = 'benchmark_layer_call__Conv2D_small_shape', the return value is ['Conv2D', 'small', 'shape']. This is to generate the metadata of the benchmark test. Args: name: A string, the benchmark name. Returns: A list of strings of the suffix in the benchmark name. ___z*The format of the benchmark name is wrong.) ValueErrorsplit)names `/home/james-whalen/.local/lib/python3.13/site-packages/tf_keras/src/benchmarks/benchmark_util.pyget_benchmark_namer s= 43d?EFF ::d B  % %c **c /nUH_nURUVs/sHo3SS-4USS-PM sn5 URUVs/sHo3SS-4USS-PM sn5 Ma U$s snfs snf)zExtend the benchmark names with CPU and GPU suffix. Args: *params_list: A list of tuples represents the benchmark parameters. Returns: A list of strings with the benchmark name extended with CPU and GPU suffix. r_CPUN_GPU)extend) params_listbenchmark_paramsparamsparams r !generate_benchmark_params_cpu_gpur,s=C DVEQx& "U12Y.V D  =C DVEQx& "U12Y.V D   E Es A* A/ c*SX-S[U5-S.$)Nkeras_examplesbs_) model_nameimplementation parameters)str) keras_model batch_sizeimpls r get_keras_examples_metadatar#As#'%,c*o- rc0\rSrSrSrSrSrSrSrSr g) TimerCallBackKz1Callback for logging time in each epoch or batch.c~/Ul[RUl[R"5UlSUlg)NF)timestimeit default_timertimer startup_timerecorded_startup)selfs r __init__TimerCallBack.__init__Ns/ )) "002 %rc.UR5UlgN)r+epoch_start_timer.elogss r on_epoch_beginTimerCallBack.on_epoch_beginTs $ rcpURRUR5UR- 5 gr2)r(appendr+r3r4s r on_epoch_endTimerCallBack.on_epoch_endWs% $**,)>)>>?rczUR(d*UR5UR- UlSUlgg)NT)r-r+r,r4s r on_batch_endTimerCallBack.on_batch_endZs1$$ $ t/@/@ @D $(D !%r)r3r-r,r+r(N) __name__ __module__ __qualname____firstlineno____doc__r/r7r;r>__static_attributes__rr r%r%Ks;& -@)rr%c Uc [S5eUc [S5eUc [S5eU S:a [S5eURSn ///pn ///nnnX<-n[R"XS9n[ U5GHKn[ R nU"5n[R"U5nU U"5nU"5nU"5U- nU"5nURUUUS9 U"5U- nSSS5 WRXUS S 9 [5nU"5nURUUUUU/U S 9 U"5n U RW5 URW5 URUR5 UR[R"UR55 URU U- 5 URUU U- - 5 GMN /nURS [R"U 5S .5 URS[R"U5S .5 URS[R"U5S .5 URS[R"U5S .5 URS[R"U5S .5 URSUS .5 [R"U5n!U U S.n"UU!U"4$!,(df  GN=f)auRun models and measure the performance. Args: model_fn: Model function to be benchmarked. x: Input data. See `x` in the `fit()` method of `keras.Model`. y: Target data. See `y` in the `fit()` method of `keras.Model`. epochs: Integer. Number of epochs to train the model. If unspecified, `epochs` will default to 2. batch_size: Integer. Number of samples per gradient update. If unspecified, `batch_size` will default to 32. run_iters: Integer. Number of iterations to run the performance measurement. If unspecified, `run_iters` will default to 4. optimizer: String (name of optimizer) or optimizer instance. See `keras.optimizers`. loss: String (name of objective function), objective function or `keras.losses.Loss` instance. See `keras.losses`. metrics: Lists of metrics to be evaluated by the model during training. See `metrics` in the `compile()` method of `keras.Model`. verbose: 0, 1, 2. Verbosity mode. See `verbose` in the `fit()` method of `keras.Model`. If unspecified, `verbose` will default to 0. num_gpus: Number of GPUs to run the model. distribution_strategy: Distribution strategies. It could be `multi_worker_mirrored`, `one_device`, `mirrored`. If unspecified, `distribution_strategy` will default to 'off'. Note that, `TPU` and `parameter_server` are not supported yet. Returns: Performance summary, which contains build_time, compile_time, startup_time, avg_epoch_time, wall_time, exp_per_sec, epochs, distribution_strategy. Raise: ValueError: If `x` is none or if `optimizer` is not provided or if `loss` is not provided or if `num_gpus` is negative. NzInput data is required.zOptimizer is required.zLoss function is required.rz`num_gpus` cannot be negative)distribution_strategynum_gpus) optimizerlossmetricsr)xyr!epochs)rMrNr!rOrverbose build_time)r value compile_timer,avg_epoch_time exp_per_secrO)r shaperget_distribution_strategyranger)r*get_strategy_scopecompilefitr%r:r,npmeanr()#model_fnrMrNrOr! run_itersrJrKrLrPrIrH num_examplesbuild_time_listcompile_time_liststartup_time_listavg_epoch_time_listwall_time_listexp_per_sec_listtotal_num_examplesstrategyrr+ start_timestrategy_scopet0modelrQt1rScbkt2end_time wall_timeextrass# r measure_performancers`sb y233  122 566 A899771:L<>B(9O<>B)9. ::3H9 $$W *==hG BJE2JB MM#   !7Rr|se# "5+&,#6)I&&).    @&r