Searched refs:reorder_attr (Results 1 – 2 of 2) sorted by relevance
98 mkldnn::primitive_attr reorder_attr; in Compute() local99 reorder_attr.set_output_scales(2, scales); in Compute()130 output_mem_prim->get_desc(), reorder_attr); in Compute()
1810 mkldnn::primitive_attr reorder_attr; in AllocateOutputTensor() local1812 reorder_attr.set_output_scales(0, scales); in AllocateOutputTensor()1814 reorder_attr.set_output_scales(2, scales); in AllocateOutputTensor()1830 conv_prim_desc.dst_desc(), reorder_attr); in AllocateOutputTensor()