Searched refs:TensorBuffer (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 70 class BufferBase : public TensorBuffer {
73 : TensorBuffer(data_ptr), alloc_(alloc) {} in BufferBase()
75 TensorBuffer* root_buffer() override { return this; } in root_buffer()
135 static void Encode(TensorBuffer* in, int64 n, Destination* out) { in Encode()
144 static TensorBuffer* Decode(Allocator* a, const Source& in, int64 n) { in Decode()
160 static int64 TotalBytes(TensorBuffer* in, int64 n) { in TotalBytes()
176 static void Encode(TensorBuffer* in, int64 n, Destination* out) { in Encode()
184 static TensorBuffer* Decode(Allocator* a, const Source& in, int64 n) { in Decode()
196 static int64 TotalBytes(TensorBuffer* in, int n) { in TotalBytes()
213 static void Encode(TensorBuffer* in, int64 n, Destination* out) { in Encode()
[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 | 44 class TensorBuffer; variable
605 TensorBuffer* buf_;
642 Tensor(DataType type, const TensorShape& shape, TensorBuffer* buf);
677 class TensorBuffer : public core::RefCounted {
679 explicit TensorBuffer(void* data_ptr) : data_(data_ptr) {} in TensorBuffer() function
680 ~TensorBuffer() override {} in ~TensorBuffer()
692 virtual TensorBuffer* root_buffer() = 0;
923 class Tensor::HostScalarTensorBufferBase : public TensorBuffer {
925 using TensorBuffer::TensorBuffer;
938 TensorBuffer* root_buffer() final { return this; } in root_buffer()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map.cc | 28 class BaseTfLiteTensorBuffer : public tensorflow::TensorBuffer {
29 using tensorflow::TensorBuffer::TensorBuffer;
31 TensorBuffer* root_buffer() override { return this; } in root_buffer()
|
| /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()
|
| D | scoped_allocator.h | 71 TensorBuffer* tbuf_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scoped_allocator_ops.cc | 126 const TensorBuffer* backing_buf = DMAHelper::buffer(&output); in Compute()
132 const TensorBuffer* input_buf = DMAHelper::buffer(&context->input(i)); in Compute()
191 const TensorBuffer* backing_buf = DMAHelper::buffer(&backing_copy); in Compute()
206 const TensorBuffer* input_buf = DMAHelper::buffer(©); in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.h | 198 class XlaTensorBuffer : public TensorBuffer {
202 : TensorBuffer(const_cast<void*>(ptr)), in XlaTensorBuffer()
215 TensorBuffer* root_buffer() override { return this; } in root_buffer()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_internal.h | 62 tensorflow::TensorBuffer* buffer;
199 static TensorBuffer* Buffer(const Tensor& tensor) { return tensor.buf_; } in Buffer()
201 TensorBuffer* buf) { in MakeTensor()
|
| D | c_api.cc | 95 using tensorflow::TensorBuffer;
142 class TF_ManagedBuffer : public TensorBuffer {
147 : TensorBuffer(data), in TF_ManagedBuffer()
161 TensorBuffer* root_buffer() override { return this; } in root_buffer()
247 TensorBuffer* buf = tensor->buffer; in TF_TensorMaybeMove()
284 static void UnrefIfNonNull(::tensorflow::TensorBuffer* buf) { in UnrefIfNonNull()
290 static void RefIfNonNull(::tensorflow::TensorBuffer* buf) { in RefIfNonNull()
590 TensorBuffer* buf = TensorCApi::Buffer(src); in TF_TensorFromTensor()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 225 const TensorBuffer* buf = DMAHelper::buffer(&val); in EncodeTensorToByteBuffer()
230 [](void* backing) { static_cast<TensorBuffer*>(backing)->Unref(); }, in EncodeTensorToByteBuffer()
231 const_cast<TensorBuffer*>(buf)); in EncodeTensorToByteBuffer()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 79 class TestTensorBuffer : public TensorBuffer {
82 : TensorBuffer(nullptr), bytes_(bytes) { in TestTensorBuffer()
90 TensorBuffer* root_buffer() override { return nullptr; } in root_buffer()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 162 std::map<TensorKey, const TensorBuffer*> tensor_buffers_ GUARDED_BY(buf_mu_);
318 const TensorBuffer* buffer = iter->second; in Run()
370 const TensorBuffer* buffer = DMAHelper::buffer(&tensor); in TransportOptionsFromTensor()
453 const TensorBuffer* buffer = DMAHelper::buffer(tensor); in TensorFromTransportOptions()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_func.cc | 305 class NumpyTensorBuffer : public TensorBuffer {
308 : TensorBuffer(data), array_(array), len_(len) {} in NumpyTensorBuffer()
317 TensorBuffer* root_buffer() override { return this; } in root_buffer()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.h | 355 TensorBuffer* src_buffer_ = nullptr;
|
| D | rdma.cc | 1203 src_buffer_ = const_cast<TensorBuffer*>(DMAHelper::buffer(&in)); in SendContent()
|