Searched refs:RewardSpec (Results 1 – 2 of 2) sorted by relevance
264 const TensorSpec RewardSpec; member in llvm::LoggerDataImpl272 const TensorSpec &RewardSpec, bool IncludeReward) in LoggerDataImpl() argument273 : 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() argument477 : FeatureSpecs(FeatureSpecs), RewardSpec(RewardSpec), in Logger()479 LoggerData(std::make_unique<LoggerDataImpl>(FeatureSpecs, RewardSpec, in Logger()[all …]
148 const TensorSpec &RewardSpec, bool IncludeReward);177 TensorSpec RewardSpec; variable