Home
last modified time | relevance | path

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

/llvm-project-15.0.7/llvm/lib/Analysis/
H A DTFUtils.cpp245 if (Spec.isElementType<float>()) { in addNewTensor()
252 } else if (Spec.isElementType<int32_t>() || Spec.isElementType<int64_t>()) { in addNewTensor()
265 if (RewardSpec.isElementType<float>()) in logReward()
267 else if (RewardSpec.isElementType<int32_t>() || in logReward()
268 RewardSpec.isElementType<int64_t>()) in logReward()
452 assert(RewardSpec.isElementType<TYPE>()); \ in LOG_REWARD()
464 assert(FeatureSpecs[FeatureID].Spec.isElementType<float>());
469 assert(FeatureSpecs[FeatureID].Spec.isElementType<int64_t>()); in logInt64Value()
474 assert(FeatureSpecs[FeatureID].Spec.isElementType<int32_t>()); in logInt32Value()
481 if (Spec.isElementType<int32_t>()) in logSpecifiedTensorValue()
[all …]
H A DTensorSpec.cpp115 if (!TensorSpec->isElementType<int64_t>() && in loadOutputSpecs()
116 !TensorSpec->isElementType<int32_t>() && in loadOutputSpecs()
117 !TensorSpec->isElementType<float>()) { in loadOutputSpecs()
/llvm-project-15.0.7/llvm/unittests/Analysis/
H A DTensorSpecTest.cpp53 EXPECT_TRUE(Spec1D.isElementType<int16_t>()); in TEST()
54 EXPECT_FALSE(Spec3DLarge.isElementType<double>()); in TEST()
/llvm-project-15.0.7/llvm/include/llvm/Analysis/
H A DTensorSpec.h80 template <typename T> bool isElementType() const { in isElementType() function