|
5 | 5 | | **RFC #** | 246 | |
6 | 6 | | **Author(s) ** | Penporn Koanantakool ( [email protected]) | |
7 | 7 | | **Sponsor ** | Rasmus Larsen ( [email protected]), Tatiana Shpeisman ( [email protected]) | |
8 | | -| **Updated** | 2019-05-19 | |
| 8 | +| **Updated** | 2019-05-28 | |
9 | 9 |
|
10 | 10 | ## Objective |
11 | 11 |
|
@@ -136,6 +136,8 @@ CSRSparseMatrix is compatible with all platforms supported by TensorFlow. |
136 | 136 |
|
137 | 137 |
|
138 | 138 | ### Tutorials and Examples |
| 139 | +Once the implementation is complete, we plan to post a tutorial on the TensorFlow blog or the TensorFlow Tutorial section. All Python and C++ APIs will be documented on the TensorFlow website just like other TensorFlow ops. |
| 140 | + |
139 | 141 | The following snippet shows how CSRSparseMatrix can be used in Python. |
140 | 142 | ```Python |
141 | 143 | import tf.linalg.experimental.sparse as csr |
@@ -191,7 +193,7 @@ D_csr = csr.concat([A_csr, B_csr, C_csr], axis=1) |
191 | 193 | * This design will conform to the backward and forward compatibility requirements once it is moved outside the experimental package. It only adds new functionalities without making changes to existing features. |
192 | 194 | * How this proposal interacts with other parts of the TensorFlow Ecosystem: |
193 | 195 | * TFLite: TFLite already supports the CSR format. TensorFlow should be able to pass the format to TFLite without problems. |
194 | | - * Distribution strategies: Don’t plan on interacting with this in this initial phase. Could overlap with DTensor. |
| 196 | + * Distribution strategies: Don’t plan on interacting with this in this initial phase. |
195 | 197 | * tf.function: Should work just like any other ops. |
196 | 198 | * GPU: We plan to make all CSRSparseMatrix operations work on GPUs. |
197 | 199 | * TPU: We don’t plan on supporting CSRSparseMatrix on TPUs yet. |
@@ -283,7 +285,7 @@ The Python [CSRSparseMatrix](https://cs.opensource.google/tensorflow/tensorflow/ |
283 | 285 |
|
284 | 286 |
|
285 | 287 | ### Shape Inference |
286 | | -`Variant tensors` are perceived as scalars in TensorFlow. For proper shape inferences, we store `CSRSparseMatrix`’s shape and data type in a shape inference primitive, [ShapeAndType](https://cs.opensource.google/tensorflow/tensorflow/+/v2.2.0-rc4:tensorflow/core/framework/shape_inference.h;l=133), and access them through [input_handle_shapes_and_types](https://cs.opensource.google/tensorflow/tensorflow/+/v2.2.0-rc4:tensorflow/core/framework/shape_inference.h;l=584) and [set_output_handle_shapes_and_types](https://cs.opensource.google/tensorflow/tensorflow/+/v2.2.0-rc4:tensorflow/core/framework/shape_inference.h;l=588) during shape inference. |
| 288 | +`Variant` tensors are perceived as scalars in TensorFlow. For proper shape inference, we store `CSRSparseMatrix`’s shape and data type in a shape inference primitive, [ShapeAndType](https://cs.opensource.google/tensorflow/tensorflow/+/v2.2.0-rc4:tensorflow/core/framework/shape_inference.h;l=133), and access them through [input_handle_shapes_and_types](https://cs.opensource.google/tensorflow/tensorflow/+/v2.2.0-rc4:tensorflow/core/framework/shape_inference.h;l=584) and [set_output_handle_shapes_and_types](https://cs.opensource.google/tensorflow/tensorflow/+/v2.2.0-rc4:tensorflow/core/framework/shape_inference.h;l=588) during shape inference. |
287 | 289 |
|
288 | 290 |
|
289 | 291 | ### APIs |
|
0 commit comments