swift-apis

swift-apis copied to clipboard

Multiple return parameters

2

I'm trying to implement a simple/limited version of the `meshgrid` op. This is what I've got: ``` func meshgrid(x: Tensor, y: Tensor) -> (Tensor, Tensor) { let outputX = x.reshaped(to:...

tanmayb123

swift build fails

4

Upon executing `swift build`, I'm receiving the following message: Any workarounds ? ``` error: found 1 file(s) which are unhandled; explicitly declare them as resources or exclude from the target...

vballoli

[x10] Add counters for operators which use interoperability

1

We just need to expose and call: https://github.com/tensorflow/swift-apis/blob/e6afd0ba29570ac285241bc92bd04be896b98396/Sources/x10/xla_client/metrics.h#L156 with the name being the operation.

asuhan

Parameter counting API for Layer

6

Currently we can't easily know how many parameters `Layer` instance has. It'll be useful for estimating model size. Keras's `Layer` has this feature. https://github.com/keras-team/keras/blob/c10d24959b0ad615a21e671b180a1b2466d77a2b/keras/engine/base_layer.py#L1105-L1123

t-ae

Add linear algebra operations and their derivatives

10

Swift for TensorFlow should provide standard linear algebra operations from [`tf.linalg`](https://www.tensorflow.org/api_docs/python/tf/linalg)/[`np.linalg`](https://docs.scipy.org/doc/numpy-1.13.0/reference/routines.linalg.html): - [x] [`tf.linalg.band_part`](https://www.tensorflow.org/api_docs/python/tf/linalg/band_part) - Done in https://github.com/tensorflow/swift-apis/pull/571. - [x] [`tf.linalg.cholesky`](https://www.tensorflow.org/api_docs/python/tf/linalg/cholesky) - Done in https://github.com/tensorflow/swift-apis/pull/563. - [ ] [`tf.linalg.cholesky_solve`](https://www.tensorflow.org/api_docs/python/tf/linalg/cholesky_solve) -...

dan-zheng

Add more optimizers and losses

16

Just a small-ish roadmap to different Optimizers and losses we can look at to add : Optimizers:- - [x] Adam - [x] Adagrad - [x] SGD - [x] RMSprop -...

Shashi456

Exposure of _ExecutionContext into public API

3

Currently I'm trying to implement custom CUDA kernel for S4TF. It seems mostly straightforward (given that C API contains TF_LibraryLoad and all other necessary stuff), but operation launching is what...

s1ddok

Add derivative tests for layers

7

Layers currently lack derivative tests. * Tensor operation derivatives are tested in `Tests/TensorFlowTests/TensorAutoDiffTests.swift`. * Layers are tested in `Tests/TensorFlowTests/LayerTests.swift`. (#77) Consider adding layer derivative tests to `Tests/TensorFlowTests/LayerTests.swift`, alongside layer "forward-pass"...

dan-zheng

Q: How to compute a gradient w.r.t. selected parameters?

12

Hello all, I have a very simple question. How to compute gradients w.r.t. a subset of parameters. Let's say I have a model like: ```swift struct Model: Layer { var...

awav

Bias shape of Conv2D

1

I found strange use of `Conv2D` in its test. https://github.com/tensorflow/swift-apis/blob/3304db3e728120b55674cca06894b6ea5083b5e8/Tests/TensorFlowTests/LayerTests.swift#L101-L111 The filter has shape `[1, 2, 2, 1]` and input has shape `[2, 2, 2, 2]`. `Conv2D` has strides `(2,...

t-ae