Home
last modified time | relevance | path

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

/netbsd-src/external/apache2/llvm/dist/llvm/include/llvm/Analysis/Utils/
H A DTFUtils.h48 class TensorSpec final {
51 static TensorSpec createSpec(const std::string &Name,
54 return TensorSpec(Name, Port, getDataType<T>(), Shape);
62 bool operator==(const TensorSpec &Other) const {
67 bool operator!=(const TensorSpec &Other) const { return !(*this == Other); }
79 TensorSpec(const std::string &Name, int Port, int TypeIndex,
100 Optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx,
104 TensorSpec Spec;
141 const TensorSpec &RewardSpec, bool IncludeReward) in Logger()
167 TensorSpec RewardSpec;
[all …]
/netbsd-src/external/apache2/llvm/dist/llvm/lib/Analysis/
H A DTFUtils.cpp171 size_t TensorSpec::getElementByteSize() const { in getElementByteSize()
175 TensorSpec::TensorSpec(const std::string &Name, int Port, int TypeIndex, in TensorSpec() function in llvm::TensorSpec
181 Optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx, in getTensorSpecFromJSON()
183 auto EmitError = [&](const llvm::Twine &Message) -> Optional<TensorSpec> { in getTensorSpecFromJSON()
212 return TensorSpec::createSpec<T>(TensorName, TensorShape, TensorPort); in getTensorSpecFromJSON()
249 if (auto TensorSpec = getTensorSpecFromJSON(Ctx, *SpecPart)) in loadOutputSpecs() local
251 if (!TensorSpec->isElementType<int64_t>() && in loadOutputSpecs()
252 !TensorSpec->isElementType<int32_t>() && in loadOutputSpecs()
253 !TensorSpec->isElementType<float>()) { in loadOutputSpecs()
257 TensorSpec->name()); in loadOutputSpecs()
[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