Method SetShapeExpectation

tensorflow/lite/testing/tflite_driver.cc:456–464

}

void TfLiteDriver::SetShapeExpectation(int id, const string& csv_values) {
  if (!IsValid()) return;
  if (expected_output_shape_.count(id) != 0) {
    Invalidate(
        absl::StrCat("Overridden shape expectation for tensor '", id, "'"));
  }
  expected_output_shape_[id].reset(new Expectation);
  expected_output_shape_[id]->SetData<int32_t>(csv_values);
}

void TfLiteDriver::Invoke() {
  if (!IsValid()) return;