Searched refs:op_scale (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 209 float op_scale = post_op_param.param[0]; in Setup() local
212 post_ops.append_eltwise(op_scale, mkldnn::algorithm::eltwise_relu, in Setup()
216 float op_scale = post_op_param.param[0]; in Setup() local
219 post_ops.append_eltwise(op_scale, in Setup()
224 float op_scale = post_op_param.param[0]; in Setup() local
227 post_ops.append_eltwise(op_scale, mkldnn::algorithm::eltwise_elu, in Setup()
231 float op_scale = post_op_param.param[0]; in Setup() local
234 post_ops.append_eltwise(op_scale, mkldnn::algorithm::eltwise_tanh, in Setup()
243 float op_scale = post_op_param.param[0]; in Setup() local
244 post_ops.append_sum(op_scale); in Setup()
|
| D | mkl_conv_ops.cc | 257 float op_scale = post_op_param.param[0]; in Setup() local
260 post_ops.append_eltwise(op_scale, post_op_param.alg, op_alpha, in Setup()
264 float op_scale = post_op_param.param[0]; in Setup() local
265 post_ops.append_sum(op_scale); in Setup()
|