Method get

compiler/circle-eval-diff/src/InputDataLoader.cpp:117–156

uint32_t HDF5Loader::size(void) const { return _hdf5->numData(); }

InputDataLoader::Data HDF5Loader::get(uint32_t data_idx) const
{
  Data data;
  data.resize(_input_nodes.size());

  for (uint32_t input_idx = 0; input_idx < _input_nodes.size(); input_idx++)
  {
    auto input_node = loco::must_cast<luci::CircleInput *>(_input_nodes.at(input_idx));
    assert(input_node->index() == input_idx);

    data.at(input_idx) = *createEmptyTensor(input_node).get();

    auto input_buffer = data.at(input_idx).buffer();
    const auto input_buffer_bytes = data.at(input_idx).byte_size();

    try
    {
      if (_hdf5->isRawData())
      {
        _hdf5->readTensor(data_idx, input_idx, input_buffer, input_buffer_bytes);
      }
      else
      {
        DataType dtype;
        Shape shape;
        _hdf5->readTensor(data_idx, input_idx, &dtype, &shape, input_buffer, input_buffer_bytes);

        // Check the type and the shape of the input data is valid
        verifyTypeShape(input_node, dtype, shape);
      }
    }
    catch (const H5::Exception &e)
    {
      H5::Exception::printErrorStack();
      throw std::runtime_error("HDF5 error occurred.");
    }
  }

  return data;
}

DirectoryLoader::DirectoryLoader(const std::string &dir_path,
                                 const std::vector<loco::Node *> &input_nodes)