Lines Matching refs:TensorSpec

32   template <> TensorType TensorSpec::getDataType<T>() { return TensorType::E; }
38 TensorSpec::TensorSpec(const std::string &Name, int Port, TensorType Type,  in SUPPORTED_TENSOR_TYPES()
45 Optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx,  in getTensorSpecFromJSON()
47   auto EmitError = [&](const llvm::Twine &Message) -> Optional<TensorSpec> {  in getTensorSpecFromJSON()
76     return TensorSpec::createSpec<T>(TensorName, TensorShape, TensorPort);  in getTensorSpecFromJSON()
113         if (auto TensorSpec = getTensorSpecFromJSON(Ctx, *SpecPart))  in loadOutputSpecs()  local
115             if (!TensorSpec->isElementType<int64_t>() &&  in loadOutputSpecs()
116                 !TensorSpec->isElementType<int32_t>() &&  in loadOutputSpecs()
117                 !TensorSpec->isElementType<float>()) {  in loadOutputSpecs()
121                   TensorSpec->name());  in loadOutputSpecs()
124             Ret.push_back({*TensorSpec, LoggingName->str()});  in loadOutputSpecs()