Searched refs:AddInputToSignatureDef (Results 1 – 3 of 3) sorted by relevance
54 internal::AddInputToSignatureDef(regression_signature.input().tensor_name(), in BuildRegressionSignatureDef()68 internal::AddInputToSignatureDef( in BuildClassificationSignatureDef()106 internal::AddInputToSignatureDef(map_entry.second.tensor_name(), in MaybeBuildPredictSignatureDef()243 void AddInputToSignatureDef( in AddInputToSignatureDef() function
38 void AddInputToSignatureDef(
115 TEST(BundleShimTest, AddInputToSignatureDef) { in TEST() argument124 AddInputToSignatureDef(tensor_name, tensor_name_to_dtype, map_key, in TEST()