record_batch_end (definition) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/trainer/
H A D	trainer.py	`66 def record_batch_end(self, key): member in TrainerBase`
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/server/
H A D	server.py	`112 def record_batch_end(self, key): member in ParameterServerBase`