IntenseWebs/openvino
3
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add mixed precision case (#11747)

Browse Source
This commit is contained in:
Sungeun Kim authored and GitHub committed 2022-06-04 11:40:52 +09:00
1 parent dba4dbb9d6
commit 8029fd9675
1 file changed
+2
Download Patch File Download Diff File Expand all files Collapse all files
src/plugins/intel_gpu/src/graph/convolution.cpp
+2
View File
@@ -182,6 +182,8 @@ layout convolution_inst::calc_output_layout(convolution_node const& node) {
out_fmt = format::b_fs_yx_fsv32;
else if (input_layout.format == format::bs_fs_yx_bsv32_fsv16)
out_fmt = format::bs_fs_yx_bsv32_fsv32;
else if (input_layout.format == format::b_fs_yx_fsv2)
out_fmt = format::b_fs_yx_fsv32;
} else if (is_3d) {
if (input_layout.format == format::b_fs_zyx_fsv16)
out_fmt = format::b_fs_zyx_fsv32;