Searched defs:allocation_attr (Results 1 – 8 of 8) sorted by relevance
102 const AllocationAttributes& allocation_attr) { in AllocateRaw()124 const AllocationAttributes& allocation_attr) { in Allocate()305 const AllocationAttributes& allocation_attr) override { in AllocateRaw()
34 const AllocationAttributes& allocation_attr) { in AllocateRaw()
450 const AllocationAttributes& allocation_attr) in Buffer()747 const AllocationAttributes& allocation_attr) in Tensor()
704 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor()745 const AllocationAttributes& allocation_attr) { in allocate_temp()
57 AllocationAttributes allocation_attr; in AllocateBytes() local
325 AllocationAttributes allocation_attr; in AllocateBytes() local
72 const AllocationAttributes& allocation_attr) { in InitPartial()
187 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry()213 const AllocationAttributes& allocation_attr) { in AllocateRaw()