experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update Max Pool 2d to support versions 1 and 2. This is accurate since we support int8 data types in the kernel implementation.

Browse Source 
PiperOrigin-RevId: 305286353
Change-Id: I026587dd5a5d59edc5484f4788e5e4bb0c75d006
This commit is contained in:
Nat Jeffries 2020-04-07 10:22:20 -07:00 committed by TensorFlower Gardener
parent aad7799759
commit bfc8b7cab0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/micro/kernels/all_ops_resolver.cc
View File

@ -22,7 +22,7 @@ namespace micro {
// AddBuiltin(<operator ID>, <registration>, [min version], [max version])
AllOpsResolver::AllOpsResolver() {
AddBuiltin(BuiltinOperator_FULLY_CONNECTED, Register_FULLY_CONNECTED(), 1, 4);
AddBuiltin(BuiltinOperator_MAX_POOL_2D, Register_MAX_POOL_2D());
AddBuiltin(BuiltinOperator_MAX_POOL_2D, Register_MAX_POOL_2D(), 1, 2);
AddBuiltin(BuiltinOperator_SOFTMAX, Register_SOFTMAX(), 1, 2);
AddBuiltin(BuiltinOperator_LOGISTIC, Register_LOGISTIC());
AddBuiltin(BuiltinOperator_SVDF, Register_SVDF(), 1, 3);