HomeSort by relevance Sort by last modified time
    Searched refs:allocated_buffer (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
scratch_allocator.cc 32 TF_ASSIGN_OR_RETURN(OwningDeviceMemory allocated_buffer,
37 se::DeviceMemoryBase buffer_addr = allocated_buffer.AsDeviceMemoryBase();
38 allocated_buffers_.push_back(std::move(allocated_buffer));
fft_thunk.cc 51 TF_ASSIGN_OR_RETURN(OwningDeviceMemory allocated_buffer,
56 se::DeviceMemoryBase buffer_addr = allocated_buffer.AsDeviceMemoryBase();
57 allocated_buffers_.push_back(std::move(allocated_buffer));
  /external/tensorflow/tensorflow/core/common_runtime/sycl/
sycl_allocator.cc 46 const auto& allocated_buffer = sycl_device_->get_sycl_buffer(p); local
47 const std::size_t bytes_allocated = allocated_buffer.get_range().size();
  /external/python/cpython3/Modules/
overlapped.c 55 PyObject *allocated_buffer; member in union:__anon37408::__anon37409
556 self->allocated_buffer = NULL;
573 Py_CLEAR(self->allocated_buffer);
705 assert(PyBytes_CheckExact(self->allocated_buffer));
706 if (transferred != PyBytes_GET_SIZE(self->allocated_buffer) &&
707 _PyBytes_Resize(&self->allocated_buffer, transferred))
709 Py_INCREF(self->allocated_buffer);
710 return self->allocated_buffer;
773 self->allocated_buffer = buf;
878 self->allocated_buffer = buf
    [all...]

Completed in 193 milliseconds