OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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