Searched refs:set_fuse_activation (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 798 void set_fuse_activation(bool fuse_activation, in set_fuse_activation() function in tensorflow::MklConvOp
1183 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_relu); in MklFusedConvOp()
1185 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_bounded_relu, in MklFusedConvOp()
1188 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_elu, 1.0); in MklFusedConvOp()
1193 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_relu, in MklFusedConvOp()
1197 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_relu); in MklFusedConvOp()
1203 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_bounded_relu, in MklFusedConvOp()
1210 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_elu, 1.0); in MklFusedConvOp()
1219 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_relu, in MklFusedConvOp()
1234 this->set_fuse_activation(true, mkldnn::algorithm::eltwise_relu); in MklFusedConvOp()
[all …]
|