Searched defs:allocator_attr (Results 1 – 4 of 4) sorted by relevance
106 const AllocationAttributes& allocator_attr) override { in AllocateRaw()
275 AllocatorAttributes allocator_attr) { in allocate_temp()545 const TensorShape& shape, const AllocatorAttributes& allocator_attr, in forward_input_or_allocate_temp()754 AllocatorAttributes allocator_attr, in allocate_temp()
978 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp()1228 AllocatorAttributes allocator_attr) { in allocate_tensor()
533 tensorflow::AllocatorAttributes allocator_attr; in TF_AllocateTemp() local