Home
last modified time | relevance | path

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

/freebsd-13.1/contrib/llvm-project/llvm/include/llvm/Analysis/Utils/
H A 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 {
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;
217 const std::vector<TensorSpec> &InputSpecs,
[all …]
/freebsd-13.1/contrib/llvm-project/llvm/lib/Analysis/
H A 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()
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()
264 const TensorSpec RewardSpec;
405 const TF_Output &Output, const TensorSpec &OutputSpec) { in checkReportAndInvalidate()
[all …]
H A 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()
H A DDevelopmentModeInlineAdvisor.cpp42 - "tensor_spec, followed by the TensorSpec description of the