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

Method transferInputToDevice

samples/common/sampleInference.cpp:1447–1456  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1445}
1446
1447void Bindings::transferInputToDevice(TrtCudaStream& stream)
1448{
1449 for (auto& b : mNames)
1450 {
1451 if (mBindings[b.second].isInput)
1452 {
1453 mBindings[b.second].buffer->hostToDevice(stream);
1454 }
1455 }
1456}
1457
1458void Bindings::transferOutputToHost(TrtCudaStream& stream)
1459{

Callers 1

setInputDataMethod · 0.80

Calls 1

hostToDeviceMethod · 0.45

Tested by

no test coverage detected