Skip to content

Commit 955a5c3

Browse files
rjauhari2kiriti-pendyala
authored andcommitted
Add data_format attr to _ZenQuantizedConv2D fuse op definitions to fix Unknown-attribute warning
- ZENAI-1799 Change-Id: I838a003384a9e62a14ab5e05cff46e3e96cfbfa2 Signed-off-by: Rahul Jauhari <rahul.jauhari@amd.com> (cherry picked from commit 8d69968)
1 parent eb36cbe commit 955a5c3

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

tensorflow_plugin/src/amd_cpu/ops/zendnn/zen_conv2d_op.cc

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
/*******************************************************************************
2-
* Copyright (c) 2022-2024 Advanced Micro Devices, Inc. All rights reserved.
2+
* Copyright (c) 2022-2025 Advanced Micro Devices, Inc. All rights reserved.
33
*
44
* Licensed under the Apache License, Version 2.0 (the "License");
55
* you may not use this file except in compliance with the License.
@@ -264,6 +264,7 @@ void RegisterZenQuantizedConv2DWithBiasAndReluAndRequantize() {
264264
TF_OpDefinitionBuilderAddOutput(op_builder, "output: quint8");
265265
TF_OpDefinitionBuilderAddOutput(op_builder, "min_output: float");
266266
TF_OpDefinitionBuilderAddOutput(op_builder, "max_output: float");
267+
TF_OpDefinitionBuilderAddAttr(op_builder, "data_format: string = 'NHWC'");
267268
TF_OpDefinitionBuilderAddAttr(op_builder, "Tinput: quantizedtype");
268269
TF_OpDefinitionBuilderAddAttr(op_builder, "Tfilter: quantizedtype");
269270
TF_OpDefinitionBuilderAddAttr(op_builder, "Tbias: {float, qint32}");
@@ -319,6 +320,7 @@ void RegisterZenQuantizedConv2DWithBiasSumAndReluAndRequantize() {
319320
TF_OpDefinitionBuilderAddOutput(op_builder, "min_output: float");
320321
TF_OpDefinitionBuilderAddOutput(op_builder, "max_output: float");
321322
TF_OpDefinitionBuilderAddAttr(op_builder, "Tinput: quantizedtype");
323+
TF_OpDefinitionBuilderAddAttr(op_builder, "data_format: string = 'NHWC'");
322324
TF_OpDefinitionBuilderAddAttr(op_builder, "Tfilter: quantizedtype");
323325
TF_OpDefinitionBuilderAddAttr(op_builder, "Tbias: {float, qint32}");
324326
TF_OpDefinitionBuilderAddAttr(op_builder, "Tsummand: quantizedtype");
@@ -373,6 +375,7 @@ void RegisterZenQuantizedConv2DWithBiasSignedSumAndReluAndRequantize() {
373375
TF_OpDefinitionBuilderAddOutput(op_builder, "output: quint8");
374376
TF_OpDefinitionBuilderAddOutput(op_builder, "min_output: float");
375377
TF_OpDefinitionBuilderAddOutput(op_builder, "max_output: float");
378+
TF_OpDefinitionBuilderAddAttr(op_builder, "data_format: string = 'NHWC'");
376379
TF_OpDefinitionBuilderAddAttr(op_builder, "Tinput: quantizedtype");
377380
TF_OpDefinitionBuilderAddAttr(op_builder, "Tfilter: quantizedtype");
378381
TF_OpDefinitionBuilderAddAttr(op_builder, "Tbias: {float, qint32}");

0 commit comments

Comments
 (0)