Searched refs:post_ops (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 204 mkldnn::post_ops post_ops; in Setup() local
212 post_ops.append_eltwise(op_scale, mkldnn::algorithm::eltwise_relu, in Setup()
219 post_ops.append_eltwise(op_scale, in Setup()
227 post_ops.append_eltwise(op_scale, mkldnn::algorithm::eltwise_elu, in Setup()
234 post_ops.append_eltwise(op_scale, mkldnn::algorithm::eltwise_tanh, in Setup()
244 post_ops.append_sum(op_scale); in Setup()
256 post_ops_attr.set_post_ops(post_ops); in Setup()
|
| D | mkl_conv_ops.cc | 252 mkldnn::post_ops post_ops; in Setup() local
260 post_ops.append_eltwise(op_scale, post_op_param.alg, op_alpha, in Setup()
265 post_ops.append_sum(op_scale); in Setup()
278 post_ops_attr.set_post_ops(post_ops); in Setup()
|