Home
last modified time | relevance | path

Searched refs:post_ops_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_quantize_op.cc136 mkldnn::primitive_attr post_ops_attr; in Setup() local
142 post_ops_attr.set_output_scales(0, scales); in Setup()
146 context_.dst_mem->get_desc(), post_ops_attr)); in Setup()
Dmkl_matmul_ops_common.h203 mkldnn::primitive_attr post_ops_attr; in Setup() local
240 post_ops_attr.set_output_scales(0, scales); in Setup()
256 post_ops_attr.set_post_ops(post_ops); in Setup()
258 *context_.fwd_desc, post_ops_attr, cpu_engine_)); in Setup()
Dmkl_conv_ops.cc251 mkldnn::primitive_attr post_ops_attr; in Setup() local
268 post_ops_attr.set_output_scales(0, post_op_param.param); in Setup()
270 post_ops_attr.set_output_scales(2, post_op_param.param); in Setup()
278 post_ops_attr.set_post_ops(post_ops); in Setup()
280 new ConvFwdPd(*context_.fwd_desc, post_ops_attr, cpu_engine_)); in Setup()