Add bfloat16 support to TileOp.

PiperOrigin-RevId: 224894043

Add bfloat16 support to TileOp.
PiperOrigin-RevId: 224894043
512f0fa9 · Russell Power · TensorFlower Gardener · f1ad9aa9 · 512f0fa9
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

tensorflow/core/kernels/tile_ops.cc tensorflow/core/kernels/tile_ops.cc +1 -0

未找到文件。
--- a/tensorflow/core/kernels/tile_ops.cc
+++ b/tensorflow/core/kernels/tile_ops.cc
@@ -325,6 +325,7 @@ class TileGradientOp : public OpKernel {
    TF_CALL_int16(HANDLE_TYPE_NAME);
    TF_CALL_int64(HANDLE_TYPE_NAME);
    TF_CALL_half(HANDLE_TYPE_NAME);
+    TF_CALL_bfloat16(HANDLE_TYPE_NAME);
    TF_CALL_complex64(HANDLE_TYPE_NAME);
    TF_CALL_complex128(HANDLE_TYPE_NAME);