HomeSort by relevance Sort by last modified time
    Searched refs:TensorBuffer (Results 1 - 12 of 12) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
dma_helper.h 29 static TensorBuffer* buffer(Tensor* t) { return t->buf_; }
30 static const TensorBuffer* buffer(const Tensor* t) { return t->buf_; }
  /external/tensorflow/tensorflow/core/framework/
tensor_reference.h 24 // An opaque class that holds a reference to an underlying TensorBuffer.
28 // TensorBuffer.
62 size_t BufferHash() const { return std::hash<TensorBuffer*>()(buf_); }
65 explicit TensorReference(TensorBuffer* test_buffer) : buf_(test_buffer) {
70 TensorBuffer* buf_;
tensor.cc 66 class BufferBase : public TensorBuffer {
70 TensorBuffer* root_buffer() override { return this; }
131 static void Encode(TensorBuffer* in, int64 n, Destination* out) {
140 static TensorBuffer* Decode(Allocator* a, const Source& in, int64 n) {
156 static int64 TotalBytes(TensorBuffer* in, int64 n) {
172 static void Encode(TensorBuffer* in, int64 n, Destination* out) {
180 static TensorBuffer* Decode(Allocator* a, const Source& in, int64 n) {
192 static int64 TotalBytes(TensorBuffer* in, int n) {
209 static void Encode(TensorBuffer* in, int64 n, Destination* out) {
217 static TensorBuffer* Decode(Allocator* a, const Source& in, int64 n)
    [all...]
tensor.h 40 class TensorBuffer;
476 TensorBuffer* buf_;
499 Tensor(DataType type, const TensorShape& shape, TensorBuffer* buf);
534 class TensorBuffer : public core::RefCounted {
536 ~TensorBuffer() override {}
542 // If this TensorBuffer is sub-buffer of another TensorBuffer,
543 // returns that TensorBuffer. Otherwise, returns this.
544 virtual TensorBuffer* root_buffer() = 0;
555 // Whether this TensorBuffer owns the underlying memory
    [all...]
  /external/tensorflow/tensorflow/c/
c_api_internal.h 58 tensorflow::TensorBuffer* buffer;
184 static TensorBuffer* Buffer(const Tensor& tensor) { return tensor.buf_; }
186 TensorBuffer* buf) {
c_api.cc 83 using tensorflow::TensorBuffer;
130 class TF_ManagedBuffer : public TensorBuffer {
143 TensorBuffer* root_buffer() override { return this; }
231 TensorBuffer* buf = tensor->buffer;
561 TensorBuffer* buf = TensorCApi::Buffer(src);
    [all...]
  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
grpc_tensor_coding.cc 217 const TensorBuffer* buf = DMAHelper::buffer(&val);
222 [](void* backing) { static_cast<TensorBuffer*>(backing)->Unref(); },
223 const_cast<TensorBuffer*>(buf));
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_event_mgr_test.cc 80 // A TensorBuffer that counts live memory usage for testing
81 class TestTensorBuffer : public TensorBuffer {
92 TensorBuffer* root_buffer() override { return nullptr; }
  /external/tensorflow/tensorflow/python/lib/core/
py_func.cc 267 class NumpyTensorBuffer : public TensorBuffer {
280 TensorBuffer* root_buffer() override { return this; }
  /external/tensorflow/tensorflow/contrib/verbs/
rdma.h 355 TensorBuffer* src_buffer_ = nullptr;
rdma.cc     [all...]
  /external/tensorflow/tensorflow/contrib/gdr/
gdr_memory_manager.cc 169 std::map<TensorKey, const TensorBuffer*> tensor_buffers_
398 const TensorBuffer* buffer = iter->second;

Completed in 331 milliseconds