def validate_and_inference_input_output(graph_def, \ input_tensor_names, output_tensor_names): from lpot.adaptor.tf_utils.util import get_input_node_names input_tensor_names = input_tensor_names if validate_graph_node(\ graph_def, tensor_to_node(input_tensor_names)) else \ get_input_node_names(graph_def) from lpot.adaptor.tf_utils.util import get_output_node_names output_tensor_names = output_tensor_names if validate_graph_node(\ graph_def, tensor_to_node(output_tensor_names)) else \ get_output_node_names(graph_def) return input_tensor_names, output_tensor_names
def validate_and_inference_input_output(graph_def, \ input_tensor_names, output_tensor_names): """validate and inference the input and output tensor names of graph_def Args: graph_def (tf.compat.v1.GraphDef): tf.compat.v1.GraphDef object input_tensor_names (list of string): input_tensor_names of graph_def output_tensor_names (list of string): output_tensor_names of graph_def Returns: input_tensor_names (list of string): validated input_tensor_names output_tensor_names (list of string): validated output_tensor_names """ from lpot.adaptor.tf_utils.util import get_input_node_names input_tensor_names = input_tensor_names if validate_graph_node(\ graph_def, tensor_to_node(input_tensor_names)) else \ get_input_node_names(graph_def) from lpot.adaptor.tf_utils.util import get_output_node_names output_tensor_names = output_tensor_names if validate_graph_node(\ graph_def, tensor_to_node(output_tensor_names)) else \ get_output_node_names(graph_def) return input_tensor_names, output_tensor_names