Searched defs:root_buffer (Results 1 – 12 of 12) sorted by relevance
41 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tensorflow::__anon0ac7c5150111::MlirTensorBuffer
84 class root_buffer : public buffer { class
79 root_buffer::root_buffer(clover::context &ctx, cl_mem_flags flags, in root_buffer() function in root_buffer
208 TensorBuffer* root_buffer() override { return this; } in root_buffer() function
74 const se::DeviceMemoryBase& root_buffer() const { in root_buffer() function
54 TensorBuffer* root_buffer() override { return this; } in root_buffer() function
32 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tflite::flex::__anonad5345df0111::BaseTfLiteTensorBuffer
87 TensorBuffer* root_buffer() override { return this; } in root_buffer() function in tensorflow::__anon8c201cc00111::BufferBase842 TensorBuffer* root_buffer() override { return root_; } in root_buffer() function in tensorflow::SubBuffer
974 TensorBuffer* root_buffer() final { return this; } in root_buffer() function
102 TensorBuffer* root_buffer() override { return nullptr; } in root_buffer() function in tensorflow::TestTensorBuffer
1842 se::DeviceMemoryBase root_buffer = root_buffer_holder.root_buffer(); in MakeOutputBuffers() local
3317 const BufferAllocation::Slice root_buffer = in GetBufferForGlobalCallReturnValue() local