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()
3625 size_t nweight = 0; in ConvertFusedBatchNorm() local3627 nweight = std::max<size_t>(nweight, inputs.at(i).weights().count()); in ConvertFusedBatchNorm()3631 if (inputs.at(i).weights().count() == nweight) { in ConvertFusedBatchNorm()3665 for (size_t i = 0; i < nweight; ++i) { in ConvertFusedBatchNorm()3699 nvinfer1::ScaleMode mode = nweight == 1 ? nvinfer1::ScaleMode::kUNIFORM in ConvertFusedBatchNorm()