Searched defs:act_layer (Results 1 – 2 of 2) sorted by relevance
70 NEActivationLayer act_layer(&ctx); in TEST_SUITE() local
719 act_layer=torch.nn.ReLU, argument