Home
last modified time | relevance | path

Searched defs:CustomTensorEvaluator (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h28 struct CustomTensorEvaluator { struct
29 typedef TensorVolumePatchOp<Planes, Rows, Cols, ArgType> XprType;
53 CustomTensorEvaluator(const XprType& op, const Device& device) in CustomTensorEvaluator() argument
239 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { in dimensions()
243 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool evalSubExprsIfNeeded( in evalSubExprsIfNeeded()
249 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void cleanup() { m_impl.cleanup(); } in cleanup()
252 coeff(Index index) const { in coeff()
334 packet(Index index) const { in packet()
434 costPerCoeff(bool vectorized) const { in costPerCoeff()
441 EIGEN_DEVICE_FUNC Scalar* data() const { return NULL; } in data()
[all …]