Searched defs:allocator_attr (Results 1 – 4 of 4) sorted by relevance
105 const AllocationAttributes& allocator_attr) override { in AllocateRaw()
282 AllocatorAttributes allocator_attr) { in allocate_temp()568 const TensorShape& shape, const AllocatorAttributes& allocator_attr, in forward_input_or_allocate_temp()777 AllocatorAttributes allocator_attr, in allocate_temp()
1015 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp()1278 AllocatorAttributes allocator_attr) { in allocate_tensor()
530 tensorflow::AllocatorAttributes allocator_attr; in TF_AllocateTemp() local