Home
last modified time | relevance | path

Searched defs:modeldef (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/fastrnns/
H A Dtest_bench.py10 def modeldef(request, net_name, executor, fuser): function
42 def test_forward(self, modeldef, benchmark): argument
45 def test_backward(self, modeldef, benchmark): argument
H A Dprofile.py24 def run_iter(modeldef): argument
H A Dbench.py83 def train_batch(modeldef): argument