Searched refs:m_block_dimensions (Results 1 – 1 of 1) sorted by relevance
366 return m_block_dimensions.TotalSize(); in blockTotalSize()371 return m_block_dimensions; in blockDimensions()390 const IndexType coord = idx * m_block_dimensions[dim]; in blockDescriptor()392 m_block_dimensions[dim]); in blockDescriptor()414 m_block_dimensions[i] = 1; in InitializeBlockDimensions()422 m_block_dimensions = m_tensor_dimensions; in InitializeBlockDimensions()441 m_block_dimensions[dim] = in InitializeBlockDimensions()445 numext::maxi(static_cast<IndexType>(1), m_block_dimensions[dim])); in InitializeBlockDimensions()454 1.0f / static_cast<float>(m_block_dimensions.rank()))); in InitializeBlockDimensions()461 m_block_dimensions[i] = in InitializeBlockDimensions()[all …]