Searched refs:AllocationAttributes (Results 1 – 16 of 16) sorted by relevance
39 struct AllocationAttributes { struct102 const AllocationAttributes& allocation_attr) { in AllocateRaw()119 return Allocate<T>(num_elements, AllocationAttributes()); in Allocate()124 const AllocationAttributes& allocation_attr) { in Allocate()305 const AllocationAttributes& allocation_attr) override { in AllocateRaw()
60 return AllocateRaw(alignment, num_bytes, AllocationAttributes()); in AllocateRaw()63 const AllocationAttributes& allocation_attr) override;
34 const AllocationAttributes& allocation_attr) { in AllocateRaw()
969 const AllocationAttributes& allocation_attr);973 AllocationAttributes()); in allocate_temp()1243 AllocationAttributes()); in allocate_tensor()1248 const AllocationAttributes& allocation_attr);
238 AllocationAttributes attr; in allocate_temp()704 AllocatorAttributes attr, const AllocationAttributes& allocation_attr) { in allocate_tensor()706 AllocationAttributes logged_attr(allocation_attr); in allocate_tensor()745 const AllocationAttributes& allocation_attr) { in allocate_temp()
109 Buffer(Allocator* a, int64 n, const AllocationAttributes& allocation_attr);450 const AllocationAttributes& allocation_attr) in Buffer()747 const AllocationAttributes& allocation_attr) in Tensor()
108 const AllocationAttributes& allocation_attr);
56 return AllocateRaw(alignment, num_bytes, AllocationAttributes()); in AllocateRaw()60 const AllocationAttributes& allocation_attr) override;87 const AllocationAttributes& allocation_attr);
102 const AllocationAttributes& allocator_attr) override { in AllocateRaw()
187 const AllocationAttributes& allocation_attr) { in AllocateRawInternalWithRetry()213 const AllocationAttributes& allocation_attr) { in AllocateRaw()
142 AllocationAttributes empty; in TempChunk()
80 const AllocationAttributes& allocation_attr);
72 const AllocationAttributes& allocation_attr) { in InitPartial()
57 AllocationAttributes allocation_attr; in AllocateBytes()
325 AllocationAttributes allocation_attr; in AllocateBytes()
175 AllocationAttributes attrs; in Allocate()