aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/tile_ops_impl.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/tile_ops_impl.h')
-rw-r--r--tensorflow/core/kernels/tile_ops_impl.h25
1 files changed, 1 insertions, 24 deletions
diff --git a/tensorflow/core/kernels/tile_ops_impl.h b/tensorflow/core/kernels/tile_ops_impl.h
index c41e4bd74b..9861717a0b 100644
--- a/tensorflow/core/kernels/tile_ops_impl.h
+++ b/tensorflow/core/kernels/tile_ops_impl.h
@@ -21,31 +21,8 @@ limitations under the License.
#include "tensorflow/core/platform/types.h"
namespace tensorflow {
-namespace functor {
-template <typename Device, typename T, int NDIM>
-struct Tile {
- void operator()(const Device& d, typename TTypes<T, NDIM>::Tensor out,
- typename TTypes<T, NDIM>::ConstTensor in,
- const Eigen::array<int32, NDIM>& broadcast_array) const {
- if (Eigen::internal::is_same<Device, Eigen::GpuDevice>::value) {
- // Use 32bit indexing to speed up the computations
- To32Bit(out).device(d) = To32Bit(in).broadcast(broadcast_array);
- } else {
- out.device(d) = in.broadcast(broadcast_array);
- }
- }
-};
-
-template <typename Device, typename T>
-struct Tile<Device, T, 0> {
- void operator()(const Device& d, typename TTypes<T, 0>::Tensor out,
- typename TTypes<T, 0>::ConstTensor in,
- const Eigen::array<int32, 0>&) const {
- // In the scalar case we simply copy the input.
- out.device(d) = in;
- }
-};
+namespace functor {
template <typename Device, typename T, int NDIM>
struct TileGrad {