Method InitAlloc

tensorflow/contrib/star/star_tensor_coding.cc:7–17

namespace tensorflow {

void StarTensorResponse::InitAlloc(
    Device* d, const AllocatorAttributes& aa) {
  Clear();
  device_ = d;
  alloc_attrs_ = aa;
  const DeviceAttributes& da = d->attributes();
  if (alloc_attrs_.on_host() || da.device_type() == "CPU") {
    on_host_ = true;
  }
  allocator_ = device_->GetAllocator(alloc_attrs_);
}

void StarTensorResponse::Clear() {
  on_host_ = false;