Searched refs:isElementType (Results 1 – 2 of 2) sorted by relevance
179 if (!TensorSpec->isElementType<int64_t>() && in loadOutputSpecs()180 !TensorSpec->isElementType<int32_t>() && in loadOutputSpecs()181 !TensorSpec->isElementType<float>()) { in loadOutputSpecs()299 if (Spec.isElementType<float>()) { in addNewTensor()306 } else if (Spec.isElementType<int32_t>() || Spec.isElementType<int64_t>()) { in addNewTensor()318 if (RewardSpec.isElementType<float>()) in logReward()320 else if (RewardSpec.isElementType<int32_t>() || in logReward()321 RewardSpec.isElementType<int64_t>()) in logReward()509 assert(FeatureSpecs[FeatureID].Spec.isElementType<float>());526 if (Spec.isElementType<int32_t>()) in logSpecifiedTensorValue()[all …]
75 template <typename T> bool isElementType() const { in isElementType() function