|
1 | 1 | /******************************************************************************* |
2 | | - * Copyright (c) 2022-2024 Advanced Micro Devices, Inc. All rights reserved. |
| 2 | + * Copyright (c) 2022-2025 Advanced Micro Devices, Inc. All rights reserved. |
3 | 3 | * |
4 | 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
5 | 5 | * you may not use this file except in compliance with the License. |
@@ -264,6 +264,7 @@ void RegisterZenQuantizedConv2DWithBiasAndReluAndRequantize() { |
264 | 264 | TF_OpDefinitionBuilderAddOutput(op_builder, "output: quint8"); |
265 | 265 | TF_OpDefinitionBuilderAddOutput(op_builder, "min_output: float"); |
266 | 266 | TF_OpDefinitionBuilderAddOutput(op_builder, "max_output: float"); |
| 267 | + TF_OpDefinitionBuilderAddAttr(op_builder, "data_format: string = 'NHWC'"); |
267 | 268 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tinput: quantizedtype"); |
268 | 269 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tfilter: quantizedtype"); |
269 | 270 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tbias: {float, qint32}"); |
@@ -319,6 +320,7 @@ void RegisterZenQuantizedConv2DWithBiasSumAndReluAndRequantize() { |
319 | 320 | TF_OpDefinitionBuilderAddOutput(op_builder, "min_output: float"); |
320 | 321 | TF_OpDefinitionBuilderAddOutput(op_builder, "max_output: float"); |
321 | 322 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tinput: quantizedtype"); |
| 323 | + TF_OpDefinitionBuilderAddAttr(op_builder, "data_format: string = 'NHWC'"); |
322 | 324 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tfilter: quantizedtype"); |
323 | 325 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tbias: {float, qint32}"); |
324 | 326 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tsummand: quantizedtype"); |
@@ -373,6 +375,7 @@ void RegisterZenQuantizedConv2DWithBiasSignedSumAndReluAndRequantize() { |
373 | 375 | TF_OpDefinitionBuilderAddOutput(op_builder, "output: quint8"); |
374 | 376 | TF_OpDefinitionBuilderAddOutput(op_builder, "min_output: float"); |
375 | 377 | TF_OpDefinitionBuilderAddOutput(op_builder, "max_output: float"); |
| 378 | + TF_OpDefinitionBuilderAddAttr(op_builder, "data_format: string = 'NHWC'"); |
376 | 379 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tinput: quantizedtype"); |
377 | 380 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tfilter: quantizedtype"); |
378 | 381 | TF_OpDefinitionBuilderAddAttr(op_builder, "Tbias: {float, qint32}"); |
|
0 commit comments