Lines Matching full:relu

17     return torch.nn.functional.relu(torch.mm(a, b))
57         time_with_torch_timer(torch_mm_relu, (a, b), string_id="torch mm + relu")
67 torch mm + relu  mean: 0.0759 ms
71 torch mm + relu  mean: 0.0316 ms
75 torch mm + relu  mean: 0.0277 ms
79 torch mm + relu  mean: 0.0290 ms
83 torch mm + relu  mean: 0.0234 ms
87 torch mm + relu  mean: 0.0322 ms
91 torch mm + relu  mean: 0.0289 ms
95 torch mm + relu  mean: 0.7896 ms
99 torch mm + relu  mean: 1.2124 ms