Searched defs:allocator_attr (Results 1 – 2 of 2) sorted by relevance
835 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp()1086 AllocatorAttributes allocator_attr) { in allocate_tensor()
499 const TensorShape& shape, const AllocatorAttributes& allocator_attr, in forward_input_or_allocate_temp()668 AllocatorAttributes allocator_attr, in allocate_temp()