internal code change
PiperOrigin-RevId: 301405619 Change-Id: Ic11eb5922c8ff27675afda3d6b69d10e4dcbae79
This commit is contained in:
parent
67d89da9b5
commit
fd1eef6965
@ -0,0 +1,50 @@
|
||||
// Copyright 2020 Google Inc. All rights reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at:
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
import TensorFlowLiteC
|
||||
|
||||
/// A delegate that uses the `Core ML` framework for performing TensorFlow Lite graph operations.
|
||||
///
|
||||
/// - Important: This is an experimental interface that is subject to change.
|
||||
public final class CoreMLDelegate: Delegate {
|
||||
/// The configuration options for the `CoreMLDelegate`.
|
||||
public let options: Options
|
||||
|
||||
// Conformance to the `Delegate` protocol.
|
||||
public private(set) var cDelegate: CDelegate
|
||||
|
||||
/// Creates a new instance configured with the given `options`.
|
||||
///
|
||||
/// - Parameters:
|
||||
/// - options: Configurations for the delegate. The default is a new instance of
|
||||
/// `CoreMLDelegate.Options` with the default configuration values.
|
||||
public init(options: Options = Options()) {
|
||||
self.options = options
|
||||
var delegateOptions = TfLiteCoreMlDelegateOptions()
|
||||
cDelegate = TfLiteCoreMlDelegateCreate(&delegateOptions)
|
||||
}
|
||||
|
||||
deinit {
|
||||
TfLiteCoreMlDelegateDelete(cDelegate)
|
||||
}
|
||||
}
|
||||
|
||||
extension CoreMLDelegate {
|
||||
/// Options for configuring the `CoreMLDelegate`.
|
||||
// TODO(b/143931022): Add preferred device support.
|
||||
public struct Options: Equatable, Hashable {
|
||||
/// Creates a new instance with the default values.
|
||||
public init() {}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user