Method SetInput

tensorflow/lite/kernels/squeeze_test.cc:51–53

  using BaseSqueezeOpModel::BaseSqueezeOpModel;

  void SetInput(std::initializer_list<float> data) {
    PopulateTensor(input_, data);
  }

  std::vector<float> GetOutput() { return ExtractVector<float>(output_); }
  std::vector<int> GetOutputShape() { return GetTensorShape(output_); }