record_batch_start (definition) in projects: aosp_15_r20

Searched defs:record_batch_start (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/trainer/
H A D	trainer.py	`53 def record_batch_start(self, key, cuda=True): member in TrainerBase`
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/server/
H A D	server.py	`99 def record_batch_start(self, key, cuda=True): member in ParameterServerBase`