Lines Matching refs:TensorSpec

99 size_t TensorSpec::getElementByteSize() const {  in getElementByteSize()
103 TensorSpec::TensorSpec(const std::string &Name, int Port, int TypeIndex, in TensorSpec() function in llvm::TensorSpec
109 Optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx, in getTensorSpecFromJSON()
111 auto EmitError = [&](const llvm::Twine &Message) -> Optional<TensorSpec> { in getTensorSpecFromJSON()
140 return TensorSpec::createSpec<T>(TensorName, TensorShape, TensorPort); in getTensorSpecFromJSON()
177 if (auto TensorSpec = getTensorSpecFromJSON(Ctx, *SpecPart)) in loadOutputSpecs() local
179 if (!TensorSpec->isElementType<int64_t>() && in loadOutputSpecs()
180 !TensorSpec->isElementType<int32_t>() && in loadOutputSpecs()
181 !TensorSpec->isElementType<float>()) { in loadOutputSpecs()
185 TensorSpec->name()); in loadOutputSpecs()
188 Ret.push_back({*TensorSpec, LoggingName->str()}); in loadOutputSpecs()
212 const std::vector<TensorSpec> &InputSpecs,
213 function_ref<TensorSpec(size_t)> GetOutputSpecs,
259 const TensorSpec &OutputSpec);
264 const TensorSpec RewardSpec;
272 const TensorSpec &RewardSpec, bool IncludeReward) in LoggerDataImpl()
334 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluatorImpl()
335 function_ref<TensorSpec(size_t)> GetOutputSpecs, size_t OutputSpecsSize, in TFModelEvaluatorImpl()
374 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator()
375 function_ref<TensorSpec(size_t)> GetOutputSpecs, size_t OutputSpecsSize, in TFModelEvaluator()
384 const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator()
385 const std::vector<TensorSpec> &OutputSpecs, in TFModelEvaluator()
405 const TF_Output &Output, const TensorSpec &OutputSpec) { in checkReportAndInvalidate()
466 template <> int TensorSpec::getDataType<T>() { return E; }
476 const TensorSpec &RewardSpec, bool IncludeReward) in Logger()