File tree Expand file tree Collapse file tree 3 files changed +7
-0
lines changed
compiler/plugin/poplar/docs Expand file tree Collapse file tree 3 files changed +7
-0
lines changed Original file line number Diff line number Diff 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:
Original file line number Diff line number Diff 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" ,
Original file line number Diff line number Diff line change 2929from tensorflow .python .ipu .ops import embedded_runtime
3030from tensorflow .python .ipu .ops import embedding_ops
3131from tensorflow .python .ipu .ops import embedding_ops_grad
32+ from tensorflow .python .ipu .ops import f8_ops
3233from tensorflow .python .ipu .ops import functional_ops
3334from tensorflow .python .ipu .ops import functional_ops_grad
3435from tensorflow .python .ipu .ops import grad_util_ops
You can’t perform that action at this time.
0 commit comments