Searched refs:input_tensor_info (Results 1 – 2 of 2) sorted by relevance
95 input_tensor_info = utils.build_tensor_info(examples)96 if input_tensor_info.dtype != types_pb2.DT_STRING:98 signature_inputs = {signature_constants.REGRESS_INPUTS: input_tensor_info}145 input_tensor_info = utils.build_tensor_info(examples)146 if input_tensor_info.dtype != types_pb2.DT_STRING:148 signature_inputs = {signature_constants.CLASSIFY_INPUTS: input_tensor_info}
86 TensorInfo input_tensor_info = in LoadAndValidateSavedModelBundle() local90 EXPECT_EQ(DT_STRING, input_tensor_info.dtype()); in LoadAndValidateSavedModelBundle()98 ValidateHalfPlusTwo(saved_model_bundle, input_tensor_info.name(), in LoadAndValidateSavedModelBundle()