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}
272 def input_tensor_info(self) -> Mapping[str, meta_graph_pb2.TensorInfo]: member in ModelHandlerV1283 return [info.name for info in self.input_tensor_info.values()]295 for tensor_info in self.input_tensor_info.values()