Searched refs:ExtraOutputsForLogging (Results 1 – 2 of 2) sorted by relevance
97 const std::vector<TensorSpec> &ExtraOutputsForLogging) in ModelUnderTrainingRunner() argument99 OutputSpecs(OutputSpecs), ExtraOutputsForLogging(ExtraOutputsForLogging) { in ModelUnderTrainingRunner()131 std::vector<TensorSpec> ExtraOutputsForLogging; in createAndEnsureValid() local136 append_range(ExtraOutputsForLogging, in createAndEnsureValid()146 Ctx, ModelPath, InputSpecs, OutputSpecs, ExtraOutputsForLogging)); in createAndEnsureValid()
39 return ExtraOutputsForLogging; in extraOutputsForLoggingSpecs() 64 const std::vector<TensorSpec> &ExtraOutputsForLogging = {});71 const std::vector<TensorSpec> ExtraOutputsForLogging;70 const std::vector<TensorSpec> ExtraOutputsForLogging; global() variable