0cbbd15f48
c_api_types.h The aim here is to remove the dependency on common.h from the C API, so that the C API depends on abstract types and enums as much as possible, without any need to include the header files that define the concrete struct types. The one non-opaque type is TFLiteQuantizationParams, which is used as a return type in the C API and therefore can't be made opaque. PiperOrigin-RevId: 347654215 Change-Id: I5cdd44624731e95b2ea477f3f86b77b18c669487
89 lines
2.7 KiB
Ruby
89 lines
2.7 KiB
Ruby
Pod::Spec.new do |s|
|
|
s.name = 'TensorFlowLiteObjC'
|
|
s.version = '2.4.0'
|
|
s.authors = 'Google Inc.'
|
|
s.license = { :type => 'Apache' }
|
|
s.homepage = 'https://github.com/tensorflow/tensorflow'
|
|
s.source = { :git => 'https://github.com/tensorflow/tensorflow.git', :tag => "v#{s.version}" }
|
|
s.summary = 'TensorFlow Lite for Objective-C'
|
|
s.description = <<-DESC
|
|
|
|
TensorFlow Lite is TensorFlow's lightweight solution for Objective-C
|
|
developers. It enables low-latency inference of on-device machine learning
|
|
models with a small binary size and fast performance supporting hardware
|
|
acceleration.
|
|
DESC
|
|
|
|
s.ios.deployment_target = '9.0'
|
|
|
|
s.module_name = 'TFLTensorFlowLite'
|
|
s.static_framework = true
|
|
|
|
tfl_dir = 'tensorflow/lite/'
|
|
objc_dir = tfl_dir + 'experimental/objc/'
|
|
|
|
s.pod_target_xcconfig = {
|
|
'HEADER_SEARCH_PATHS' =>
|
|
'"${PODS_TARGET_SRCROOT}" ' +
|
|
'"${PODS_TARGET_SRCROOT}/' + objc_dir + 'apis"',
|
|
'VALID_ARCHS' => 'i386 x86_64 armv7 arm64',
|
|
}
|
|
|
|
s.default_subspec = 'Core'
|
|
|
|
s.subspec 'Core' do |core|
|
|
core.public_header_files = objc_dir + 'apis/*.h'
|
|
core.source_files = [
|
|
objc_dir + '{apis,sources}/*.{h,m,mm}',
|
|
tfl_dir + 'c/c_api.h',
|
|
tfl_dir + 'c/c_api_types.h',
|
|
tfl_dir + 'c/common.h',
|
|
tfl_dir + 'delegates/xnnpack/xnnpack_delegate.h',
|
|
]
|
|
core.exclude_files = [
|
|
objc_dir + '{apis,sources}/TFL{Metal,CoreML}Delegate.{h,m}',
|
|
]
|
|
core.dependency 'TensorFlowLiteC', "#{s.version}"
|
|
|
|
core.test_spec 'Tests' do |ts|
|
|
ts.source_files = objc_dir + 'tests/*.m'
|
|
ts.exclude_files = objc_dir + 'tests/TFL{Metal,CoreML}DelegateTests.m'
|
|
ts.resources = [
|
|
tfl_dir + 'testdata/add.bin',
|
|
tfl_dir + 'testdata/add_quantized.bin',
|
|
]
|
|
end
|
|
end
|
|
|
|
s.subspec 'CoreML' do |coreml|
|
|
coreml.source_files = [
|
|
objc_dir + '{apis,sources}/TFLCoreMLDelegate.{h,m}',
|
|
]
|
|
coreml.ios.deployment_target = '12.0'
|
|
coreml.dependency 'TensorFlowLiteC/CoreML', "#{s.version}"
|
|
coreml.dependency 'TensorFlowLiteObjC/Core', "#{s.version}"
|
|
|
|
coreml.test_spec 'Tests' do |ts|
|
|
ts.source_files = objc_dir + 'tests/TFLCoreMLDelegateTests.m'
|
|
ts.resources = [
|
|
tfl_dir + 'testdata/add.bin',
|
|
]
|
|
end
|
|
end
|
|
|
|
s.subspec 'Metal' do |metal|
|
|
metal.source_files = [
|
|
objc_dir + '{apis,sources}/TFLMetalDelegate.{h,m}',
|
|
]
|
|
metal.dependency 'TensorFlowLiteC/Metal', "#{s.version}"
|
|
metal.dependency 'TensorFlowLiteObjC/Core', "#{s.version}"
|
|
|
|
metal.test_spec 'Tests' do |ts|
|
|
ts.source_files = objc_dir + 'tests/TFLMetalDelegateTests.m'
|
|
ts.resources = [
|
|
tfl_dir + 'testdata/multi_add.bin',
|
|
]
|
|
end
|
|
end
|
|
end
|