| /external/mesa3d/src/gallium/frontends/clover/core/ |
| D | memory.cpp | 88 root_buffer::root_buffer(clover::context &ctx, in root_buffer() function in root_buffer
96 root_buffer::resource_in(command_queue &q) { in resource_in()
105 root_buffer::resource_undef(command_queue &q) { in resource_undef()
110 root_buffer::resource(command_queue &q, const void *data_ptr) { in resource()
128 root_buffer::resource_out(command_queue &q) { in resource_out()
133 sub_buffer::sub_buffer(root_buffer &parent, cl_mem_flags flags, in sub_buffer()
|
| D | memory.hpp | 90 class root_buffer : public buffer { class
92 root_buffer(clover::context &ctx,
115 sub_buffer(root_buffer &parent, cl_mem_flags flags,
126 const intrusive_ref<root_buffer> parent;
|
| D | printf.cpp | 92 _buffer = std::unique_ptr<root_buffer>(new root_buffer(_q->context, in printf_handler()
|
| D | error.hpp | 46 class root_buffer;
145 class invalid_object_error<root_buffer> : public error {
|
| D | printf.hpp | 56 std::unique_ptr<root_buffer> _buffer;
|
| D | kernel.hpp | 255 std::map<device *, std::unique_ptr<root_buffer> > _constant_buffers;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_reference.h | 34 : buf_(tensor.buf_ ? tensor.buf_->root_buffer() : nullptr) { in TensorReference()
|
| D | tensor.cc | 92 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tensorflow::__anon2f56f8af0111::BufferBase
768 buf_->root_buffer()->RefCountIsOne() && buf_->OwnsMemory(); in RefCountIsOne()
892 root_(buf->root_buffer()), in SubBuffer()
905 TensorBuffer* root_buffer() override { return root_; } in root_buffer() function in tensorflow::SubBuffer
1323 buf_->root_buffer() == b.buf_->root_buffer(); in SharesBufferWith()
|
| D | tensor.h | 82 virtual TensorBuffer* root_buffer() = 0;
1013 TensorBuffer* root_buffer() final { return this; } in root_buffer() function
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map_util.h | 30 inline TensorBuffer* root_buffer() override { return this; } in root_buffer() function
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | fallback_tensor.cc | 45 tensorflow::TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tensorflow::tfrt_stub::__anonf555a6100111::ImmutableTensorBuffer
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_op.cc | 41 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tensorflow::__anon9f1157440111::MlirTensorBuffer
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt.h | 54 TensorBuffer* root_buffer() override { return this; } in root_buffer() function
|
| D | tf_jitrt.h | 63 TensorBuffer* root_buffer() override { return this; } in root_buffer() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 37 return xla_tensor->shaped_buffer().root_buffer(); in DeviceMemoryFromTensor()
|
| D | xla_launch_util.h | 216 TensorBuffer* root_buffer() override { return this; } in root_buffer() function
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor_internal.h | 55 TensorBuffer* root_buffer() override { return this; } in root_buffer() function
|
| D | tf_tensor.cc | 207 if (buf->RefCountIsOne() && buf->root_buffer()->RefCountIsOne() && in CanMove()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shaped_buffer.h | 73 const se::DeviceMemoryBase& root_buffer() const { in root_buffer() function
|
| /external/mesa3d/src/intel/vulkan/grl/gpu/ |
| D | bvh_build_BFS.cl | 4061 global uint2* root_buffer = (global uint2*) globals->p_qnode_root_buffer;
4077 root_buffer[get_sub_group_local_id()] = children_info.xy;
4104 uniform uint bvh2_node = root_buffer[buffer_index].x;
4105 uniform uint qnode_index = root_buffer[buffer_index].y;
4133 root_buffer[root_buffer_position + get_sub_group_local_id()] = children_info.xy;
4364 global uint2* root_buffer,
4396 …amplify_and_spill( globals, alloc_backpointers, qnode_root, bvh2_root, root_buffer, SLM_spill_pos,…
4402 root_buffer[i] = SLM_local_root_buffer[(ring_head + i) % RING_SIZE];
4424 global uint2* root_buffer,
4439 root_buffer,
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 236 *run_options, memory_bases, result.Result().root_buffer(), in ExecuteAsyncOnStream()
|
| /external/mesa3d/src/gallium/frontends/clover/api/ |
| D | memory.cpp | 124 return new root_buffer(ctx, properties, flags, size, host_ptr); in clCreateBufferWithProperties()
142 auto &parent = obj<root_buffer>(d_mem); in clCreateSubBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 142 se::DeviceMemoryBase result_root_buffer = output.Result().root_buffer(); in RunAndCheck()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 2076 se::DeviceMemoryBase root_buffer = root_buffer_holder.root_buffer(); in MakeOutputBuffers() local
2078 [root_buffer, allocator{client_->allocator()}, device_ordinal]() { in MakeOutputBuffers()
2079 TF_CHECK_OK(allocator->Deallocate(device_ordinal, root_buffer)); in MakeOutputBuffers()
|
| /external/mesa3d/src/panfrost/lib/genxml/ |
| D | cs_builder.h | 200 struct cs_buffer root_buffer) in cs_builder_init() argument
204 .root_chunk.buffer = root_buffer, in cs_builder_init()
205 .cur_chunk.buffer = root_buffer, in cs_builder_init()
|