44584be99d
Added protection from NULL delegate PiperOrigin-RevId: 333668953 Change-Id: Id1b8c09d4c8b830814261fa4f64c0850aa7a2f46
88 lines
2.7 KiB
Plaintext
88 lines
2.7 KiB
Plaintext
Pod::Spec.new do |s|
|
|
s.name = 'TensorFlowLiteObjC'
|
|
s.version = '${TFL_BUILD_VERSION}'
|
|
s.authors = 'Google Inc.'
|
|
s.license = { :type => 'Apache' }
|
|
s.homepage = 'https://github.com/tensorflow/tensorflow'
|
|
s.source = { :git => 'https://github.com/tensorflow/tensorflow.git', :commit => '${SOURCE_COMMIT}' }
|
|
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/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
|