Home
last modified time | relevance | path

Searched refs:apply_input_props_using_example (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/jit/_passes/
D_property_propagation.py16 def apply_input_props_using_example(graph: Graph, example_input: List[Any]): function
/external/pytorch/test/jit/
Dtest_device_analysis.py7 from torch.jit._passes._property_propagation import apply_input_props_using_example
98 apply_input_props_using_example(graph, in_example)
Dtest_dtype_analysis.py123 _property_propagation.apply_input_props_using_example(graph, example_inputs)