MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / Execute

Method Execute

src/opr/impl/nvof/NvOF.cpp:126–146  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

124}
125
126void NvOF::Execute(
127 NvOFBuffer* image1, NvOFBuffer* image2, NvOFBuffer* outputBuffer,
128 NvOFBuffer* hintBuffer, NvOFBuffer* costBuffer) {
129 NV_OF_EXECUTE_INPUT_PARAMS exeInParams;
130 NV_OF_EXECUTE_OUTPUT_PARAMS exeOutParams;
131
132 memset(&exeInParams, 0, sizeof(exeInParams));
133 exeInParams.inputFrame = image1->getOFBufferHandle();
134 exeInParams.referenceFrame = image2->getOFBufferHandle();
135 exeInParams.disableTemporalHints = NV_OF_FALSE;
136 exeInParams.externalHints = m_initParams.enableExternalHints == NV_OF_TRUE
137 ? hintBuffer->getOFBufferHandle()
138 : nullptr;
139
140 memset(&exeOutParams, 0, sizeof(exeOutParams));
141 exeOutParams.outputBuffer = outputBuffer->getOFBufferHandle();
142 exeOutParams.outputCostBuffer = m_initParams.enableOutputCost == NV_OF_TRUE
143 ? costBuffer->getOFBufferHandle()
144 : nullptr;
145 DoExecute(exeInParams, exeOutParams);
146}
147
148std::vector<std::unique_ptr<NvOFBuffer>> NvOF::CreateBuffers(
149 NV_OF_BUFFER_USAGE usage, uint32_t numBuffers) {

Callers 1

extract_flowMethod · 0.80

Calls 1

getOFBufferHandleMethod · 0.80

Tested by

no test coverage detected