Skip to content

Instantly share code, notes, and snippets.

@r7vme

r7vme/enqueue.cpp

Created July 8, 2019 22:20
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save r7vme/c27880cdd78ba193a9630d2c6012e1e5 to your computer and use it in GitHub Desktop.
Save r7vme/c27880cdd78ba193a9630d2c6012e1e5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
enqueue.cpp
int L2NormHelper::enqueue(int batchSize, const void* const* inputs, void** outputs, void* workspace, cudaStream_t stream)
{
const void* inputData = inputs[0];
void* outputData = outputs[0];
bool status = executeInference(stream, op_type, eps, batchSize, C, H, W, inputData, outputData);
ASSERT(status == 0);
return 0;
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment