Searched refs:alloc_attrs_ (Results 1 – 3 of 3) sorted by relevance
31 alloc_attrs_ = AllocatorAttributes(); in Clear()45 alloc_attrs_ = aa; in InitAlloc()47 if (alloc_attrs_.on_host() || da.device_type() == "CPU") { in InitAlloc()50 allocator_ = device_->GetAllocator(alloc_attrs_); in InitAlloc()61 s = device_->MakeTensorFromProto(meta_.tensor(), alloc_attrs_, &tensor_); in InitFrom()92 device_->MakeTensorFromProto(meta_.tensor(), alloc_attrs_, &tensor_); in ParseFrom()
102 AllocatorAttributes alloc_attrs_; variable
65 alloc_attrs_ = alloc_attrs; in Init()80 alloc_attrs_ = AllocatorAttributes(); in Reset()138 resp_.InitAlloc(dst_device_, alloc_attrs_); in StartRTCall()157 AllocatorAttributes alloc_attrs_; member in tensorflow::__anon9a114d310111::RpcRecvTensorCall