tf CategoryEncoding layer support

[jeffkrupa]

Hi, I am using experimental preprocessing layer CategoryEncoding in tensorflow 2.4.1. Its operations are not supported (below). Is support for this planned?

Thank you.

System information

• OS Platform and Distribution: centos 7
• Tensorflow Version: 2.4.1
• Python version: 3.8.8

Error

2021-03-16 11:59:41,208 - WARNING - '--tag' not specified for saved_model. Using --tag serve
2021-03-16 11:59:42,146 - INFO - Signatures found in model: [serving_default].
2021-03-16 11:59:42,146 - WARNING - '--signature_def' not specified, using first signature: serving_default
WARNING:tensorflow: extract_sub_graph (from tensorflow.python.framework.graph_util_impl) is deprecated and will be removed in a future version.
Instructions for updating:
Use `tf.compat.v1.graph_util.extract_sub_graph`
2021-03-16 11:59:42,362 - WARNING - From: extract_sub_graph (from tensorflow.python.framework.graph_util_impl) is deprecated and will be removed in a future version.
Instructions for updating:
Use `tf.compat.v1.graph_util.extract_sub_graph`
2021-03-16 11:59:42,431 - INFO - Using tensorflow=2.4.1, onnx=1.8.1, tf2onnx=1.8.3/0fbdb5
2021-03-16 11:59:42,432 - INFO - Using opset <onnx, 9>
2021-03-16 11:59:42,440 - WARNING - Cannot infer shape for StatefulPartitionedCall/model/category_encoding_1/bincount/DenseBincount: StatefulPartitionedCall/model/category_encoding_1/bincount/DenseBincount:0
2021-03-16 11:59:42,440 - WARNING - Cannot infer shape for StatefulPartitionedCall/model/category_encoding/bincount/DenseBincount: StatefulPartitionedCall/model/category_encoding/bincount/DenseBincount:0
2021-03-16 11:59:42,452 - INFO - Computed 0 values for constant folding
2021-03-16 11:59:42,552 - ERROR - Tensorflow op [StatefulPartitionedCall/model/category_encoding/bincount/DenseBincount: DenseBincount] is not supported
2021-03-16 11:59:42,558 - ERROR - Tensorflow op [StatefulPartitionedCall/model/category_encoding_1/bincount/DenseBincount: DenseBincount] is not supported
2021-03-16 11:59:42,560 - ERROR - Unsupported ops: Counter({'DenseBincount': 2})

[TomWildenhain-Microsoft]

I should be able to implement it but there are a few settings that change the op's behavior. Can you give me the code you are using for the layer so I can cover your use case?

[jeffkrupa]

Hi, thanks for your response. I'm just using it as in

from tensorflow.keras.layers.experimental import preprocessing
encoder = preprocessing.CategoryEncoding(output_mode="binary")