Searched refs:TensorType (Results 1 – 2 of 2) sorted by relevance
31 template <> TensorType TensorSpec::getDataType<T>() { return TensorType::E; }37 static std::array<std::string, static_cast<size_t>(TensorType::Total)> in SUPPORTED_TENSOR_TYPES()44 StringRef toString(TensorType TT) { in toString()60 TensorSpec::TensorSpec(const std::string &Name, int Port, TensorType Type, in TensorSpec()85 std::string TensorType; in getTensorSpecFromJSON() local90 if (!Mapper.map<std::string>("type", TensorType)) in getTensorSpecFromJSON()98 if (TensorType == #T) \ in getTensorSpecFromJSON()
46 enum class TensorType { enum65 TensorType type() const { return Type; } in type()93 TensorSpec(const std::string &Name, int Port, TensorType Type,96 template <typename T> static TensorType getDataType();100 TensorType Type = TensorType::Invalid;117 template <> TensorType TensorSpec::getDataType<T>();