Method RunImpl

dali/operators/decoder/numpy.cc:150–165

void NumpyDecoder::RunImpl(Workspace &ws) {
  const auto &input = ws.Input<CPUBackend>(0);
  auto &output = ws.Output<CPUBackend>(0);
  auto &tPool = ws.GetThreadPool();

  for (int sampleIdx = 0; sampleIdx < input.num_samples(); ++sampleIdx) {
    tPool.AddWork([&, sampleIdx](int) {
      const numpy::HeaderData &header = headers_[sampleIdx];
      ConstSampleView<CPUBackend> inputView(
          static_cast<const uint8_t *>(input.raw_tensor(sampleIdx)) + header.data_offset,
          header.shape, header.type());
      RunDecoding(output[sampleIdx], std::move(inputView), header);
    });
  }
  tPool.RunAll();
}

DALI_REGISTER_OPERATOR(decoders__Numpy, NumpyDecoder, CPU);