MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / skip

Method skip

samples/common/BatchStream.h:238–252  ·  view source on GitHub ↗

Skips the batches

Source from the content-addressed store, hash-verified

236
237 // Skips the batches
238 void skip(int skipCount) override
239 {
240 if (mBatchSize >= mDims.d[0] && mBatchSize % mDims.d[0] == 0 && mFileBatchPos == mDims.d[0])
241 {
242 mFileCount += skipCount * mBatchSize / mDims.d[0];
243 return;
244 }
245
246 int x = mBatchCount;
247 for (int i = 0; i < skipCount; i++)
248 {
249 next();
250 }
251 mBatchCount = x;
252 }
253
254 float* getBatch() override
255 {

Callers 13

test_resnet18Method · 0.45
test_aspMethod · 0.45
test_api_examplesFunction · 0.45
test_cli_examplesFunction · 0.45
test_trt_sanityMethod · 0.45
test_flagsMethod · 0.45
test_calibrator_basicMethod · 0.45

Calls

no outgoing calls

Tested by 13

test_resnet18Method · 0.36
test_aspMethod · 0.36
test_api_examplesFunction · 0.36
test_cli_examplesFunction · 0.36
test_trt_sanityMethod · 0.36
test_flagsMethod · 0.36
test_calibrator_basicMethod · 0.36