Searched refs:max_batch_size (Results 1 – 4 of 4) sorted by relevance
70 idws->max_batch_size = 1 * 4096; in i915_drm_winsys_create()75 idws->gem_manager = drm_intel_bufmgr_gem_init(idws->fd, idws->max_batch_size); in i915_drm_winsys_create()
26 size_t max_batch_size; member
59 batch->actual_size = idws->max_batch_size; in i915_drm_batchbuffer_create()
368 max_batch_size = x.shape[0] if self.batch_first else x.shape[1]371 hx = _init_state((self.num_layers * num_directions, max_batch_size, self.hidden_size),