Lines Matching refs:RewardSpec
187 const TensorSpec RewardSpec; member in llvm::LoggerDataImpl
207 .at(RewardSpec.name()) in isSelfConsistent()
219 (*FL)[RewardSpec.name()] = std::move(Reward); in transferLog()
229 const TensorSpec &RewardSpec, bool IncludeReward) in LoggerDataImpl() argument
230 : LoggedFeatureSpecs(LoggedSpecs), RewardSpec(RewardSpec), in LoggerDataImpl()
265 if (RewardSpec.isElementType<float>()) in logReward()
267 else if (RewardSpec.isElementType<int32_t>() || in logReward()
268 RewardSpec.isElementType<int64_t>()) in logReward()
431 const TensorSpec &RewardSpec, bool IncludeReward) in Logger() argument
432 : FeatureSpecs(FeatureSpecs), RewardSpec(RewardSpec), in Logger()
434 LoggerData(std::make_unique<LoggerDataImpl>(FeatureSpecs, RewardSpec, in Logger()
452 assert(RewardSpec.isElementType<TYPE>()); \ in LOG_REWARD()