Searched refs:buffer_if_can_stack_allocate (Results 1 – 1 of 1) sorted by relevance
1580 self, name, device, dtype, shape, stride, buffer_if_can_stack_allocate=None argument1602 if buffer_if_can_stack_allocate is not None:1603 self.stack_allocated_buffers[name] = buffer_if_can_stack_allocate1605 numel = buffer_if_can_stack_allocate.get_numel()