Searched refs:reorder_attr (Results 1 – 2 of 2) sorted by relevance
93 mkldnn::primitive_attr reorder_attr; in Compute() local94 reorder_attr.set_output_scales(2, scales); in Compute()128 mkldnn::reorder::primitive_desc(input_pd, output_pd, reorder_attr); in Compute()
1818 mkldnn::primitive_attr reorder_attr; in AllocateOutputTensor() local1819 reorder_attr.set_output_scales(0, scales); in AllocateOutputTensor()1834 summand_pd, conv_prim_desc.dst_primitive_desc(), reorder_attr); in AllocateOutputTensor()