Searched refs:TensorBuffer (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 66 class BufferBase : public TensorBuffer {
70 TensorBuffer* root_buffer() override { return this; } in root_buffer()
131 static void Encode(TensorBuffer* in, int64 n, Destination* out) { in Encode()
140 static TensorBuffer* Decode(Allocator* a, const Source& in, int64 n) { in Decode()
156 static int64 TotalBytes(TensorBuffer* in, int64 n) { in TotalBytes()
172 static void Encode(TensorBuffer* in, int64 n, Destination* out) { in Encode()
180 static TensorBuffer* Decode(Allocator* a, const Source& in, int64 n) { in Decode()
192 static int64 TotalBytes(TensorBuffer* in, int n) { in TotalBytes()
209 static void Encode(TensorBuffer* in, int64 n, Destination* out) { in Encode()
217 static TensorBuffer* Decode(Allocator* a, const Source& in, int64 n) { in Decode()
[all …]
|
| D | tensor_reference.h | 62 size_t BufferHash() const { return std::hash<TensorBuffer*>()(buf_); } in BufferHash()
65 explicit TensorReference(TensorBuffer* test_buffer) : buf_(test_buffer) { in TensorReference()
70 TensorBuffer* buf_;
|
| D | tensor.h | 40 class TensorBuffer; variable
476 TensorBuffer* buf_;
499 Tensor(DataType type, const TensorShape& shape, TensorBuffer* buf);
534 class TensorBuffer : public core::RefCounted {
536 ~TensorBuffer() override {} in ~TensorBuffer()
544 virtual TensorBuffer* root_buffer() = 0;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | dma_helper.h | 29 static TensorBuffer* buffer(Tensor* t) { return t->buf_; } in buffer()
30 static const TensorBuffer* buffer(const Tensor* t) { return t->buf_; } in buffer()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_internal.h | 58 tensorflow::TensorBuffer* buffer;
184 static TensorBuffer* Buffer(const Tensor& tensor) { return tensor.buf_; } in Buffer()
186 TensorBuffer* buf) { in MakeTensor()
|
| D | c_api.cc | 83 using tensorflow::TensorBuffer;
130 class TF_ManagedBuffer : public TensorBuffer {
143 TensorBuffer* root_buffer() override { return this; } in root_buffer()
231 TensorBuffer* buf = tensor->buffer; in TF_TensorMaybeMove()
561 TensorBuffer* buf = TensorCApi::Buffer(src); in TF_TensorFromTensor()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 217 const TensorBuffer* buf = DMAHelper::buffer(&val); in EncodeTensorToByteBuffer()
222 [](void* backing) { static_cast<TensorBuffer*>(backing)->Unref(); }, in EncodeTensorToByteBuffer()
223 const_cast<TensorBuffer*>(buf)); in EncodeTensorToByteBuffer()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 81 class TestTensorBuffer : public TensorBuffer {
92 TensorBuffer* root_buffer() override { return nullptr; } in root_buffer()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_func.cc | 267 class NumpyTensorBuffer : public TensorBuffer {
280 TensorBuffer* root_buffer() override { return this; } in root_buffer()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 169 std::map<TensorKey, const TensorBuffer*> tensor_buffers_
398 const TensorBuffer* buffer = iter->second; in Run()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.h | 355 TensorBuffer* src_buffer_ = nullptr;
|
| D | rdma.cc | 1201 src_buffer_ = const_cast<TensorBuffer*>(DMAHelper::buffer(&in)); in SendContent()
|