Internal change
PiperOrigin-RevId: 573254750
This commit is contained in:
parent
1bd800697e
commit
8993073f35
|
@ -30,7 +30,7 @@ limitations under the License.
|
|||
namespace mediapipe {
|
||||
namespace tasks {
|
||||
namespace core {
|
||||
// TODO: Use separate builtin op-resolvers.
|
||||
|
||||
MediaPipeBuiltinOpResolver::MediaPipeBuiltinOpResolver() {
|
||||
AddCustom("MaxPoolingWithArgmax2D",
|
||||
mediapipe::tflite_operations::RegisterMaxPoolingWithArgmax2D());
|
||||
|
@ -60,6 +60,7 @@ MediaPipeBuiltinOpResolver::MediaPipeBuiltinOpResolver() {
|
|||
AddCustom("FusedBatchNormV3",
|
||||
mediapipe::tflite_operations::Register_FusedBatchNorm());
|
||||
}
|
||||
|
||||
} // namespace core
|
||||
} // namespace tasks
|
||||
} // namespace mediapipe
|
||||
|
|
Loading…
Reference in New Issue
Block a user