@@ -260,7 +260,7 @@ static const std::vector<TensorSpec> TrainingOnlyFeatures{
260260
261261static const std::vector<TensorSpec> getInputFeatures () {
262262 std::vector<TensorSpec> InputSpecs;
263- for (const auto &Feature : FeatureMap )
263+ for (const auto &Feature : getFeatureMap () )
264264 InputSpecs.push_back (TensorSpec (TFFeedPrefix + Feature.name (), Feature));
265265 append_range (InputSpecs, TrainingOnlyFeatures);
266266 return InputSpecs;
@@ -272,7 +272,7 @@ TrainingLogger::TrainingLogger(StringRef LogFileName,
272272 const ModelUnderTrainingRunner *MUTR)
273273 : LogFileName(LogFileName), MUTR(MUTR) {
274274 // The first output is the inlining decision.
275- std::vector<TensorSpec> FT (FeatureMap .begin (), FeatureMap .end ());
275+ std::vector<TensorSpec> FT (getFeatureMap () .begin (), getFeatureMap () .end ());
276276
277277 if (MUTR)
278278 append_range (FT, MUTR->extraOutputsForLoggingSpecs ());
@@ -298,7 +298,7 @@ void TrainingLogger::logInlineEvent(const InlineEvent &Event,
298298 const MLModelRunner &ModelRunner) {
299299 L->startObservation ();
300300 size_t CurrentFeature = 0 ;
301- size_t FeatureMapSize = FeatureMap .size ();
301+ size_t FeatureMapSize = getFeatureMap () .size ();
302302 for (; CurrentFeature < FeatureMapSize; ++CurrentFeature)
303303 L->logTensorValue (CurrentFeature,
304304 reinterpret_cast <const char *>(
0 commit comments