Searched defs:weight_count (Results 1 – 5 of 5) sorted by relevance
31 size_t TotalBytesForWeightCount(size_t weight_count) { in TotalBytesForWeightCount()89 const size_t weight_count = static_cast<size_t>(ceil(fabs(scale))) + 1; in CalculateWeights() local
121 int weight_count = ixtab2[bsd->block_modes[block_mode].decimation_mode]->num_weights; in physical_to_symbolic() local
236 int weight_count = it->num_weights; in decompress_symbolic_block() local
126 int weight_count = N * M * (D + 1); in decode_block_mode_2d() local215 int weight_count = N * M * Q * (D + 1); in decode_block_mode_3d() local
176 static void VerifyFusedMatrixClose(int depth, int batch, int weight_count, in VerifyFusedMatrixClose()