Searched defs:allocation_attr (Results 1 – 13 of 13) sorted by relevance
132 const AllocationAttributes& allocation_attr) { in AllocateRaw()255 const AllocationAttributes& allocation_attr) override { in AllocateRaw()
38 const AllocationAttributes& allocation_attr) { in Allocate()
34 const AllocationAttributes& allocation_attr) { in AllocateRaw()
491 const AllocationAttributes& allocation_attr) in Buffer()831 const AllocationAttributes& allocation_attr) in Tensor()
752 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor()823 const AllocationAttributes& allocation_attr) { in allocate_temp()
66 AllocationAttributes allocation_attr; in AllocateBytes() local
385 AllocationAttributes allocation_attr; in AllocateBytes() local
72 const AllocationAttributes& allocation_attr) { in InitPartial()
282 AllocationAttributes allocation_attr; in SameWorkerRecvDone() local
334 AllocationAttributes allocation_attr; in MaybeCopyTensorToPluggableDevice() local
233 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry()259 const AllocationAttributes& allocation_attr) { in AllocateRaw()
955 const AllocationAttributes& allocation_attr) override { in AllocateRaw()
786 AllocationAttributes allocation_attr; in MaybeCopyTensorToGPU() local