swift-apis
swift-apis copied to clipboard
Swift for TensorFlow Deep Learning Library
Preconditions (for shape-checking, valid value ranges, etc) would be valuable for better source locations in error messages and to avoid uninformative TensorFlow runtime errors. Example: `matmul(_:_:)` currently crashes with an...
It seems reduction functions are supporting axes with duplication like `[0, 0]` (`_Raw.*` handle it under the hood). But `_vjp*` don't consider it, so gradient computation can be incorrect or...
As pointed out in PR #579, the tf.keras implementation of `categorical_hinge()` expects values to be -1 or 1, but [will convert binary labels of 0 or 1 to -1 and...
I have noticed that whenever the input to the function doesn't satisfy the precondition then it result in crashing of a Google Colab and Swift-Jupyter Lab. Here is the reference...
I'm trying to use `_Raw.histogramSummary` but find problems of `StringTensor`. ```swift import TensorFlow let tag = StringTensor("tag") print("tag:", tag) print("tag.scalars:", tag.scalars) let values = Tensor(randomNormal: [100]) let summary: StringTensor =...
We should identify the actual size and performance implications of the `-sil-inline-generics` and `-sil-partial-specialization` flags that we set during the compilation here: https://github.com/tensorflow/swift-apis/blob/f0748c4771ebe33f70a531808937cbc1e58812d8/CMakeLists.txt#L11-L12
Compilation time (`swift build`) is very slow: ```console $ time swift build # clean build [61/61] Linking libTensorFlow.dylib swift build 506.63s user 5.63s system 253% cpu 3:22.07 total $ echo...
Currently, `tensorflow/swift-apis` is cloned as an `apple/swift` build product and is built as part of the Swift standard library. ([relevant CMake support](https://github.com/apple/swift/blob/tensorflow/stdlib/public/TensorFlow/CMakeLists.txt)) However, we would like to move `TensorFlow` out...
#522 cc @eaplatanios please comment if you would like to take this on eventually, and I will assign it to you.
#519 cc @Shashi456, if you want to work on this please add a comment so that I can assign it to you!