Searched refs:post_op_param (Results 1 – 1 of 1) sorted by relevance
251 for (auto const& post_op_param : post_op_params) { in Setup() local252 if (post_op_param.name == "relu") { in Setup()253 DCHECK_EQ(post_op_param.param.size(), 3); in Setup()254 float op_scale = post_op_param.param[0]; in Setup()255 float op_alpha = post_op_param.param[1]; in Setup()256 float op_beta = post_op_param.param[2]; in Setup()259 } else if (post_op_param.name == "sum") { in Setup()260 DCHECK_EQ(post_op_param.param.size(), 1); in Setup()261 float op_scale = post_op_param.param[0]; in Setup()263 } else if (post_op_param.name == "output_scale") { in Setup()[all …]