Lines Matching refs:TensorSpec
25 TensorSpec Spec;
60 if (auto TensorSpec = getTensorSpecFromJSON(Ctx, *SpecPart)) in loadOutputSpecs() local
62 if (!TensorSpec->isElementType<int64_t>() && in loadOutputSpecs()
63 !TensorSpec->isElementType<int32_t>() && in loadOutputSpecs()
64 !TensorSpec->isElementType<float>()) { in loadOutputSpecs()
68 TensorSpec->name()); in loadOutputSpecs()
71 Ret.push_back({*TensorSpec, LoggingName->str()}); in loadOutputSpecs()
95 const std::vector<TensorSpec> &InputSpecs, in ModelUnderTrainingRunner()
96 const std::vector<TensorSpec> &OutputSpecs, in ModelUnderTrainingRunner()
97 const std::vector<TensorSpec> &ExtraOutputsForLogging) in ModelUnderTrainingRunner()
125 const std::vector<TensorSpec> &InputSpecs, in createAndEnsureValid()
130 std::vector<TensorSpec> OutputSpecs; in createAndEnsureValid()
131 std::vector<TensorSpec> ExtraOutputsForLogging; in createAndEnsureValid()
139 return TensorSpec(LFS.LoggingName in createAndEnsureValid()