Skip to content

Commit 3b857e2

Browse files
Merge changes from github.
END_PUBLIC --- Commit cf375f067 authored by A. Unique TensorFlower<[email protected]> Committed by Benoit Steiner<[email protected]>: Adds cudnn_rnn_ops_op_lib and cudnn_rnn_kernels to contrib_ops_op_lib and contrib_kernels respectively. PiperOrigin-RevId: 164170971 --- Commit 95ec58e27 authored by Asim Shankar<[email protected]> Committed by Benoit Steiner<[email protected]>: C API: Make TF_TensorFromTensor return an error instead of just logging it. PiperOrigin-RevId: 164167582 --- Commit 15175c870 authored by Jonathan Hseu<[email protected]> Committed by Benoit Steiner<[email protected]>: Build fixes. - Allow var_list as a positional argument in CrossShardOptimizer. - Set the number of shards to 1 when not running on TPU, to allow evaluate() and predict() on CPU/GPU to work. PiperOrigin-RevId: 164161640 --- Commit bd3e894f7 authored by Yao Zhang<[email protected]> Committed by Benoit Steiner<[email protected]>: Support freeze mode for fused batch norm. PiperOrigin-RevId: 164149032 --- Commit e6b6b84c0 authored by Asim Shankar<[email protected]> Committed by Benoit Steiner<[email protected]>: C API: TF_Tensors will always be in host memory. This change undoes some experimentation in commit 22651083406ca01ac9d481e3367a3510d25f88cd and restores TF_Tensor behavior to what is was prior to that change. PiperOrigin-RevId: 164146670 --- Commit 8bf3f88f7 authored by Peter Hawkins<[email protected]> Committed by Benoit Steiner<[email protected]>: [TF:XLA] Add _XLASend and _XLARecv TF ops that wrap the XLA Send/Recv HLO ops. PiperOrigin-RevId: 164124764 --- Commit 626d3200f authored by Peter Hawkins<[email protected]> Committed by Benoit Steiner<[email protected]>: [XLA] Add test blacklist mechanism for XLA C++ unit tests. PiperOrigin-RevId: 164124423 --- Commit 359cc5f5e authored by A. Unique TensorFlower<[email protected]> Committed by Benoit Steiner<[email protected]>: Document dict ordering in nest and make it consistent with sonnet. PiperOrigin-RevId: 164114335 --- Commit 05813b531 authored by A. Unique TensorFlower<[email protected]> Committed by Benoit Steiner<[email protected]>: Go: Update generated wrapper functions for TensorFlow ops. PiperOrigin-RevId: 164089206 --- Commit c451f465d authored by Anna R<[email protected]> Committed by Benoit Steiner<[email protected]>: BEGIN_PUBLIC Automated g4 rollback of changelist 164078808 PiperOrigin-RevId: 164318935
1 parent ff644bd commit 3b857e2

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

extend/adding_an_op.md

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -178,9 +178,7 @@ suggested implementation is to:
178178
file, but the specialization for the GPUDevice is defined in a .cu.cc file,
179179
since it will be compiled with the CUDA compiler.
180180
181-
<!--zippy-->
182-
183-
Expand this to see the example implementation.
181+
Here is an example implementation.
184182
185183
```c++
186184
// example.h
@@ -307,8 +305,6 @@ template struct ExampleFunctor<GPUDevice, int32>;
307305
#endif // GOOGLE_CUDA
308306
```
309307

310-
<!--endzippy-->
311-
312308
## Build the op library
313309
### Compile the op using your system compiler (TensorFlow binary installation)
314310

@@ -763,7 +759,7 @@ Your op registration now specifies that the input's type must be `float`, or
763759
> """
764760
> ```
765761
766-
<pre><pre class="prettyprint"><code class="lang-cpp">
762+
<pre class="prettyprint"><code class="lang-cpp">
767763
\#include "tensorflow/core/framework/op_kernel.h"<br/>
768764
class ZeroOut<b>Int32</b>Op : public OpKernel {
769765
// as before
@@ -803,7 +799,7 @@ REGISTER\_KERNEL\_BUILDER(
803799
.Device(DEVICE\_CPU)
804800
.TypeConstraint&lt;float&gt;("T"),
805801
ZeroOutFloatOp);
806-
</b></code></pre></pre>
802+
</b></code></pre>
807803
808804
> To preserve [backwards compatibility](#backwards-compatibility), you should
809805
> specify a [default value](#default-values-constraints) when adding an attr to

0 commit comments

Comments
 (0)