index bcd61931662cc89339ecf54493cdf5d534222ad0..d7decc6b79a085a34b60d55efc147a9e92a4fb08 100644 (file)
image_file = input_file;
}
- // Determine input frame size from configuration
- size_t frame_sz = configuration.inWidth * configuration.inHeight *
- configuration.inNumChannels;
-
try
{
// Create a executor with the approriate core type, number of cores
std::vector<void *> buffers;
for (auto &eo : execution_objects)
{
- ArgInfo in = { ArgInfo(malloc(frame_sz), frame_sz)};
- ArgInfo out = { ArgInfo(malloc(frame_sz), frame_sz)};
+ size_t in_size = eo->GetInputBufferSizeInBytes();
+ size_t out_size = eo->GetOutputBufferSizeInBytes();
+ ArgInfo in = { ArgInfo(malloc(in_size), in_size)};
+ ArgInfo out = { ArgInfo(malloc(out_size), out_size)};
eo->SetInputOutputBuffer(in, out);
buffers.push_back(in.ptr());