Searched refs:TensorType (Results 1 – 2 of 2) sorted by relevance
33 template <> TensorType TensorSpec::getDataType<T>() { return TensorType::E; }39 static std::array<std::string, static_cast<size_t>(TensorType::Total)> in SUPPORTED_TENSOR_TYPES()46 StringRef toString(TensorType TT) { in toString()62 TensorSpec::TensorSpec(const std::string &Name, int Port, TensorType Type, in TensorSpec()87 std::string TensorType; in getTensorSpecFromJSON() local92 if (!Mapper.map<std::string>("type", TensorType)) in getTensorSpecFromJSON()100 if (TensorType == #T) \ in getTensorSpecFromJSON()110 case TensorType::N: { \ in tensorValueToString()118 case TensorType::Total: in tensorValueToString()119 case TensorType::Invalid: in tensorValueToString()
55 enum class TensorType { enum74 TensorType type() const { return Type; } in type()102 LLVM_ABI TensorSpec(const std::string &Name, int Port, TensorType Type,105 template <typename T> static TensorType getDataType();109 TensorType Type = TensorType::Invalid;130 template <> LLVM_ABI TensorType TensorSpec::getDataType<T>();