tensorflow/core/common_runtime/gpu/dma_helper.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_DMA_HELPER_H_
#define TENSORFLOW_COMMON_RUNTIME_GPU_DMA_HELPER_H_

#include "tensorflow/core/public/tensor.h"

// For internal use only.  Visibility should be limited to brain/framework.

namespace tensorflow {
class DMAHelper {
 public:
  static bool CanUseDMA(const Tensor* t) { return t->CanUseDMA(); }
  static const void* base(const Tensor* t) { return t->base<const void>(); }
  static void* base(Tensor* t) { return t->base<void>(); }
  static TensorBuffer* buffer(Tensor* t) { return t->buf_; }
  static const TensorBuffer* buffer(const Tensor* t) { return t->buf_; }
};
}  // namespace tensorflow
#endif  // TENSORFLOW_COMMON_RUNTIME_GPU_DMA_HELPER_H_