Home
last modified time | relevance | path

Searched refs:TensorSpec (Results 1 – 4 of 4) sorted by relevance

/freebsd-12-stable/contrib/llvm-project/llvm/include/llvm/Analysis/Utils/
DTFUtils.h49 class TensorSpec final {
52 static TensorSpec createSpec(const std::string &Name,
55 return TensorSpec(Name, Port, getDataType<T>(), Shape);
63 bool operator==(const TensorSpec &Other) const {
68 bool operator!=(const TensorSpec &Other) const { return !(*this == Other); }
80 TensorSpec(const std::string &Name, int Port, int TypeIndex,
101 Optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx,
105 TensorSpec Spec;
148 const TensorSpec &RewardSpec, bool IncludeReward);
177 TensorSpec RewardSpec;
[all …]
/freebsd-12-stable/contrib/llvm-project/llvm/lib/Analysis/
DTFUtils.cpp99 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()
[all …]
DInlineSizeEstimatorAnalysis.cpp224 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in InlineSizeEstimatorAnalysis()
228 std::vector<TensorSpec> OutputSpecs{ in InlineSizeEstimatorAnalysis()
229 TensorSpec::createSpec<float>("StatefulPartitionedCall", {1})}; in InlineSizeEstimatorAnalysis()
DDevelopmentModeInlineAdvisor.cpp42 - "tensor_spec, followed by the TensorSpec description of the