Lines Matching refs:RewardSpec
264 const TensorSpec RewardSpec; member in llvm::LoggerDataImpl
272 const TensorSpec &RewardSpec, bool IncludeReward) in LoggerDataImpl() argument
273 : LoggedFeatureSpecs(LoggedSpecs), RewardSpec(RewardSpec) { in LoggerDataImpl()
276 Reward = &(*FL)[RewardSpec.name()]; in LoggerDataImpl()
318 if (RewardSpec.isElementType<float>()) in logReward()
320 else if (RewardSpec.isElementType<int32_t>() || in logReward()
321 RewardSpec.isElementType<int64_t>()) in logReward()
476 const TensorSpec &RewardSpec, bool IncludeReward) in Logger() argument
477 : FeatureSpecs(FeatureSpecs), RewardSpec(RewardSpec), in Logger()
479 LoggerData(std::make_unique<LoggerDataImpl>(FeatureSpecs, RewardSpec, in Logger()
497 assert(RewardSpec.isElementType<TYPE>()); \ in LOG_REWARD()