[TensorSpec] Avoid JSON.h include (NFC)

Instead forward declare the two classes that are referenced.

Author: nikic

llvm/include/llvm/Analysis/TensorSpec.h

@@ -13,13 +13,17 @@
 
 #include "llvm/ADT/StringMap.h"
 #include "llvm/IR/LLVMContext.h"
-#include "llvm/Support/JSON.h"
 
 #include <memory>
 #include <optional>
 #include <vector>
 
 namespace llvm {
+namespace json {
+class OStream;
+class Value;
+} // namespace json
+
 /// TensorSpec encapsulates the specification of a tensor: its dimensions, or
 /// "shape" (row-major), its type (see TensorSpec::getDataType specializations
 /// for supported types), its name and port (see "TensorFlow: Large-Scale

llvm/lib/CodeGen/MLRegAllocEvictAdvisor.h

@@ -17,6 +17,7 @@
 #include "llvm/Analysis/MLModelRunner.h"
 #include "llvm/CodeGen/MachineBasicBlock.h"
 #include "llvm/CodeGen/SlotIndexes.h"
+#include <map>
 
 using namespace llvm;
 

llvm/lib/LTO/ThinLTOCodeGenerator.cpp

@@ -45,6 +45,7 @@
 #include "llvm/Support/Debug.h"
 #include "llvm/Support/Error.h"
 #include "llvm/Support/FileSystem.h"
+#include "llvm/Support/FormatVariadic.h"
 #include "llvm/Support/Path.h"
 #include "llvm/Support/SHA1.h"
 #include "llvm/Support/SmallVectorMemoryBuffer.h"

llvm/unittests/Analysis/TensorSpecTest.cpp

@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "llvm/Analysis/TensorSpec.h"
+#include "llvm/Support/JSON.h"
 #include "llvm/Support/Path.h"
 #include "llvm/Support/SourceMgr.h"
 #include "llvm/Testing/Support/SupportHelpers.h"
@@ -65,4 +66,4 @@ TEST(TensorSpecTest, PrintValueForDebug) {
   EXPECT_EQ(tensorValueToString(reinterpret_cast<const char *>(Values.data()),
                                 TensorSpec::createSpec<int32_t>("name", {2})),
             "1,3");
-}
+}