Home
last modified time | relevance | path

Searched refs:torch_c_binding_in_graph_functions (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_trace_rules.py20 torch_c_binding_in_graph_functions,
341 set(torch_c_binding_in_graph_functions.keys())
381 torch_c_binding_in_graph_functions,
417 torch_c_binding_in_graph_functions,
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py323 torch_c_binding_in_graph_functions = dict.fromkeys( variable
2214 torch_c_binding_in_graph_functions["math.lcm"] = TorchInGraphFunctionVariable
2216 torch_c_binding_in_graph_functions["math.exp2"] = TorchInGraphFunctionVariable
2217 torch_c_binding_in_graph_functions["math.cbrt"] = TorchInGraphFunctionVariable
2844 torch_c_binding_in_graph_functions,