Searched refs:get_input_specs (Results 1 – 1 of 1) sorted by relevance
672 def get_input_specs(graph_module: fx.GraphModule) -> Set[TensorSpec]: function698 input_specs = get_input_specs(graph_module)