Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/
H A Dlauncher.py161 def call_trainers(args, extra_args, train_data, server_rrefs): argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py131 def test_only_train_fn(model, train_data, loss_fn=_default_loss_fn): argument