| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator.cc | 126 if (ref_buffer_.root_buffer().size() != test_buffer.size()) { in CompareEqualImpl()
128 ref_buffer_.root_buffer().size(), test_buffer.size()); in CompareEqualImpl()
151 CHECK(result_buffer.root_buffer().size() == sizeof(result)); in CompareEqualImpl()
152 stream_->ThenMemcpy(&result, result_buffer.root_buffer(), sizeof(result)); in CompareEqualImpl()
167 stream_->ThenMemcpy(host_ref_buffer.data(), ref_buffer_.root_buffer(), in CompareEqual()
168 ref_buffer_.root_buffer().size()); in CompareEqual()
|
| /external/mesa3d/src/gallium/state_trackers/clover/core/ |
| D | memory.hpp | 80 class root_buffer : public buffer { class
82 root_buffer(clover::context &ctx, cl_mem_flags flags,
94 sub_buffer(root_buffer &parent, cl_mem_flags flags,
100 const intrusive_ref<root_buffer> parent;
|
| D | memory.cpp | 79 root_buffer::root_buffer(clover::context &ctx, cl_mem_flags flags, in root_buffer() function in root_buffer
85 root_buffer::resource(command_queue &q) { in resource()
101 sub_buffer::sub_buffer(root_buffer &parent, cl_mem_flags flags, in sub_buffer()
|
| D | error.hpp | 41 class root_buffer;
140 class invalid_object_error<root_buffer> : public error {
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_reference.cc | 21 : buf_(tensor.buf_ ? tensor.buf_->root_buffer() : nullptr) { in TensorReference()
|
| D | tensor_reference.h | 58 return buf_ == (t.buf_ ? t.buf_->root_buffer() : nullptr); in SharesBufferWith()
|
| D | tensor.cc | 75 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tensorflow::__anon2ce1c7b90111::BufferBase
684 buf_->root_buffer()->RefCountIsOne() && buf_->OwnsMemory(); in RefCountIsOne()
777 root_(buf->root_buffer()), in SubBuffer()
790 TensorBuffer* root_buffer() override { return root_; } in root_buffer() function in tensorflow::SubBuffer
1183 buf_->root_buffer() == b.buf_->root_buffer(); in SharesBufferWith()
|
| D | tensor.h | 692 virtual TensorBuffer* root_buffer() = 0;
938 TensorBuffer* root_buffer() final { return this; } in root_buffer() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 39 return xla_tensor->shaped_buffer().root_buffer(); in DeviceMemoryFromTensor()
|
| D | xla_launch_util.h | 215 TensorBuffer* root_buffer() override { return this; } in root_buffer() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shaped_buffer.h | 72 const se::DeviceMemoryBase& root_buffer() const { in root_buffer() function
|
| /external/mesa3d/src/gallium/state_trackers/clover/api/ |
| D | memory.cpp | 89 return new root_buffer(ctx, flags, size, host_ptr); in clCreateBuffer()
100 auto &parent = obj<root_buffer>(d_mem); in clCreateSubBuffer()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map.cc | 31 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tflite::flex::__anona9db58f80111::BaseTfLiteTensorBuffer
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 205 se::DeviceMemoryBase root_buffer = tuple_buffer.root_buffer(); in DestructureTuple() local
206 TF_RETURN_IF_ERROR(allocator->Deallocate(device_ordinal, root_buffer)); in DestructureTuple()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 90 TensorBuffer* root_buffer() override { return nullptr; } in root_buffer() function in tensorflow::TestTensorBuffer
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_func.cc | 317 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tensorflow::NumpyTensorBuffer
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 161 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in __anonef860cf70111::TF_ManagedBuffer
248 if (buf->RefCountIsOne() && buf->root_buffer()->RefCountIsOne() && in TF_TensorMaybeMove()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 3100 const BufferAllocation::Slice root_buffer = in GetBufferForGlobalCallReturnValue() local
3102 return EmitBufferPointer(root_buffer, root_inst->shape()); in GetBufferForGlobalCallReturnValue()
|