Searched refs:spatial_threads (Results 1 – 1 of 1) sorted by relevance
112 const int spatial_threads = conv_params.GetSpatialThreadsCount(); in GenerateConvolution() local113 c += "#define SPATIAL_THREADS " + std::to_string(spatial_threads) + "\n"; in GenerateConvolution()128 const bool cache_weight = spatial_threads > 32; in GenerateConvolution()171 if (spatial_threads > 32) { in GenerateConvolution()217 const int groups = weights_flt4_per_spatial / spatial_threads; in GenerateConvolution()218 const int reminder = weights_flt4_per_spatial % spatial_threads; in GenerateConvolution()220 c += " tmp_w_x4[spatial_id + " + std::to_string(spatial_threads * i) + in GenerateConvolution()222 std::to_string(spatial_threads * i) + "];\n"; in GenerateConvolution()227 std::to_string(spatial_threads * groups) + in GenerateConvolution()229 std::to_string(spatial_threads * groups) + "];\n"; in GenerateConvolution()[all …]