blob: 831f0c899edb5a0b31b15d63b4dabf03ab3e656b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
#ifndef TENSORFLOW_KERNELS_FILL_FUNCTOR_H_
#define TENSORFLOW_KERNELS_FILL_FUNCTOR_H_
#include "tensorflow/core/framework/tensor_types.h"
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
namespace tensorflow {
namespace functor {
template <typename Device, typename T>
struct FillFunctor {
// Computes on device "d": out = out.constant(in(0)),
void operator()(const Device& d, typename TTypes<T>::Flat out,
typename TTypes<T>::ConstScalar in);
};
template <typename Device, typename T>
struct SetZeroFunctor {
// Computes on device "d": out = out.setZero(),
void operator()(const Device& d, typename TTypes<T>::Flat out);
};
} // namespace functor
} // namespace tensorflow
#endif // TENSORFLOW_KERNELS_FILL_FUNCTOR_H_
|