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

Function reformat

samples/sampleIOFormats/sampleIOFormats.cpp:540–565  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

538//!
539template <typename T>
540void reformat(SampleBuffer& src, SampleBuffer& dst)
541{
542 if (src.format == dst.format)
543 {
544 memcpy(dst.buffer, src.buffer, src.getBufferSize());
545 return;
546 }
547
548 int32_t srcIndex, dstIndex;
549
550 T* srcBuf = reinterpret_cast<T*>(src.buffer);
551 T* dstBuf = reinterpret_cast<T*>(dst.buffer);
552
553 for (int32_t c = 0; c < src.dims.d[1]; c++)
554 {
555 for (int32_t h = 0; h < src.dims.d[2]; h++)
556 {
557 for (int32_t w = 0; w < src.dims.d[3]; w++)
558 {
559 srcIndex = calcIndex(src, c, h, w);
560 dstIndex = calcIndex(dst, c, h, w);
561 dstBuf[dstIndex] = srcBuf[srcIndex];
562 }
563 }
564 }
565}
566
567template <typename T>
568void convertGoldenData(SampleBuffer& goldenInput, SampleBuffer& dstInput)

Callers

nothing calls this directly

Calls 2

calcIndexFunction · 0.85
getBufferSizeMethod · 0.45

Tested by

no test coverage detected