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 mediapipe {
|
||||||
namespace tasks {
|
namespace tasks {
|
||||||
namespace core {
|
namespace core {
|
||||||
// TODO: Use separate builtin op-resolvers.
|
|
||||||
MediaPipeBuiltinOpResolver::MediaPipeBuiltinOpResolver() {
|
MediaPipeBuiltinOpResolver::MediaPipeBuiltinOpResolver() {
|
||||||
AddCustom("MaxPoolingWithArgmax2D",
|
AddCustom("MaxPoolingWithArgmax2D",
|
||||||
mediapipe::tflite_operations::RegisterMaxPoolingWithArgmax2D());
|
mediapipe::tflite_operations::RegisterMaxPoolingWithArgmax2D());
|
||||||
|
@ -60,6 +60,7 @@ MediaPipeBuiltinOpResolver::MediaPipeBuiltinOpResolver() {
|
||||||
AddCustom("FusedBatchNormV3",
|
AddCustom("FusedBatchNormV3",
|
||||||
mediapipe::tflite_operations::Register_FusedBatchNorm());
|
mediapipe::tflite_operations::Register_FusedBatchNorm());
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace core
|
} // namespace core
|
||||||
} // namespace tasks
|
} // namespace tasks
|
||||||
} // namespace mediapipe
|
} // namespace mediapipe
|
||||||
|
|
Loading…
Reference in New Issue
Block a user