Searched defs:allocator_attr (Results 1 – 3 of 3) sorted by relevance
102 const AllocationAttributes& allocator_attr) override { in AllocateRaw()
971 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp()1241 AllocatorAttributes allocator_attr) { in allocate_tensor()
566 const TensorShape& shape, const AllocatorAttributes& allocator_attr, in forward_input_or_allocate_temp()744 AllocatorAttributes allocator_attr, in allocate_temp()