Skip to content

Commit cf0511c

Browse files
committed
Adding f8_ops to api documentation
Reviewers: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, #tech_docs, jamiep, zigmasb Reviewed By: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, #tech_docs, jamiep, zigmasb Subscribers: grahamh Maniphest Tasks: T65634 Differential Revision: https://phabricator.sourcevertex.net/D78924
1 parent 27f83e7 commit cf0511c

File tree

3 files changed

+7
-0
lines changed

3 files changed

+7
-0
lines changed

tensorflow/compiler/plugin/poplar/docs/api.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,11 @@ It is also possible to access the operators via the
212212
:imported-members:
213213
:exclude-members: mul, reduce
214214

215+
.. automodule:: tensorflow.python.ipu.f8_ops
216+
:members:
217+
:imported-members:
218+
:exclude-members: canonicalise_input, canonicalise_output
219+
215220
.. automodule:: tensorflow.python.ipu.reduce_scatter_op
216221
:members:
217222
:imported-members:

tensorflow/python/ipu/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ py_library(
2828
"ops/cross_replica_ops.py",
2929
"ops/embedded_runtime.py",
3030
"ops/embedding_ops.py",
31+
"ops/f8_ops.py",
3132
"ops/functional_ops.py",
3233
"ops/grad_util_ops.py",
3334
"ops/image_ops.py",

tensorflow/python/ipu/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
from tensorflow.python.ipu.ops import embedded_runtime
3030
from tensorflow.python.ipu.ops import embedding_ops
3131
from tensorflow.python.ipu.ops import embedding_ops_grad
32+
from tensorflow.python.ipu.ops import f8_ops
3233
from tensorflow.python.ipu.ops import functional_ops
3334
from tensorflow.python.ipu.ops import functional_ops_grad
3435
from tensorflow.python.ipu.ops import grad_util_ops

0 commit comments

Comments
 (0)