Searched refs:batch_round_up (Results 1 – 1 of 1) sorted by relevance
600 int batch_round_up = n_batch; in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate() local602 batch_round_up += (4 - n_batch % 4); in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()604 TFLITE_CHECK_LE(n_batch, batch_round_up); in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()607 const int padded_vectors_size = batch_round_up * m_cols; in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()614 const int padded_result_size = batch_round_up * m_rows * sizeof(float); in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()626 const int padded_scaling_factors_size = batch_round_up * sizeof(float); in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()631 TFLITE_CHECK_LE(batch_round_up * sizeof(float), padded_scaling_factors_size); in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()632 memset(padded_scaling_factors, 0, batch_round_up * sizeof(float)); in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()637 const int padded_input_offset_size = batch_round_up * sizeof(int32_t); in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()642 TFLITE_CHECK_LE(batch_round_up * sizeof(int32_t), padded_input_offset_size); in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()[all …]