Searched refs:nweight (Results 1 – 2 of 2) sorted by relevance
481 …ize(sf). Note that if sf[i].weight is not set, the default value for the\nweight is assumed to be … in TEST_F()
5222 size_t nweight = 0; in ConvertFusedBatchNorm() local5224 nweight = std::max<size_t>(nweight, inputs.at(i).weights().count()); in ConvertFusedBatchNorm()5228 if (inputs.at(i).weights().count() == nweight) { in ConvertFusedBatchNorm()5261 for (size_t i = 0; i < nweight; ++i) { in ConvertFusedBatchNorm()5295 nvinfer1::ScaleMode mode = nweight == 1 ? nvinfer1::ScaleMode::kUNIFORM in ConvertFusedBatchNorm()