Searched refs:m_tensor_dimensions (Results 1 – 1 of 1) sorted by relevance
356 : m_tensor_dimensions(dimensions), m_requirements(requirements) { in TensorBlockMapper()391 dimensions[dim] = numext::mini(m_tensor_dimensions[dim] - coord, in blockDescriptor()406 IndexType tensor_size = m_tensor_dimensions.TotalSize(); in InitializeBlockDimensions()422 m_block_dimensions = m_tensor_dimensions; in InitializeBlockDimensions()442 numext::mini(coeff_to_allocate, m_tensor_dimensions[dim]); in InitializeBlockDimensions()462 numext::mini(dim_size_target, m_tensor_dimensions[i]); in InitializeBlockDimensions()470 if (m_block_dimensions[dim] < m_tensor_dimensions[dim]) { in InitializeBlockDimensions()480 numext::mini(m_tensor_dimensions[dim], alloc_avail); in InitializeBlockDimensions()491 m_tensor_dimensions.TotalSize())); in InitializeBlockDimensions()496 block_count[i] = divup(m_tensor_dimensions[i], m_block_dimensions[i]); in InitializeBlockDimensions()[all …]