Searched refs:post_op_param (Results 1 – 2 of 2) sorted by relevance
206 for (auto const& post_op_param : post_op_params) { in Setup()207 if (post_op_param.name == "relu" || post_op_param.name == "leakyrelu") { in Setup()208 DCHECK_EQ(post_op_param.param.size(), 3); in Setup()209 float op_scale = post_op_param.param[0]; in Setup()210 float op_alpha = post_op_param.param[1]; in Setup()211 float op_beta = post_op_param.param[2]; in Setup()214 } else if (post_op_param.name == "relu6") { in Setup()215 DCHECK_EQ(post_op_param.param.size(), 3); in Setup()216 float op_scale = post_op_param.param[0]; in Setup()217 float op_alpha = post_op_param.param[1]; in Setup()[all …]
254 for (auto const& post_op_param : post_op_params) { in Setup() local255 if (post_op_param.name == "activation") { in Setup()256 DCHECK_EQ(post_op_param.param.size(), 3); in Setup()257 float op_scale = post_op_param.param[0]; in Setup()258 float op_alpha = post_op_param.param[1]; in Setup()259 float op_beta = post_op_param.param[2]; in Setup()260 post_ops.append_eltwise(op_scale, post_op_param.alg, op_alpha, in Setup()262 } else if (post_op_param.name == "sum") { in Setup()263 DCHECK_EQ(post_op_param.param.size(), 1); in Setup()264 float op_scale = post_op_param.param[0]; in Setup()[all …]