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()
54 enum class TensorType { enum73 TensorType type() const { return Type; } in type()101 TensorSpec(const std::string &Name, int Port, TensorType Type,104 template <typename T> static TensorType getDataType();108 TensorType Type = TensorType::Invalid;128 template <> TensorType TensorSpec::getDataType<T>();