Searched refs:RewardSpec (Results 1 – 2 of 2) sorted by relevance
187 const TensorSpec RewardSpec; member in llvm::LoggerDataImpl207 .at(RewardSpec.name()) in isSelfConsistent()219 (*FL)[RewardSpec.name()] = std::move(Reward); in transferLog()229 const TensorSpec &RewardSpec, bool IncludeReward) in LoggerDataImpl() argument230 : 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() argument432 : FeatureSpecs(FeatureSpecs), RewardSpec(RewardSpec), in Logger()[all …]
75 const TensorSpec &RewardSpec, bool IncludeReward);112 TensorSpec RewardSpec; variable