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()
38 return ExtraOutputsForLogging; in extraOutputsForLoggingSpecs()63 const std::vector<TensorSpec> &ExtraOutputsForLogging = {});70 const std::vector<TensorSpec> ExtraOutputsForLogging; variable