Searched refs:create_functionalized_rng_ops_wrapper (Results 1 – 2 of 2) sorted by relevance
285 def create_functionalized_rng_ops_wrapper(func, args, trace_joint=True) -> Any: function649 helper, args = create_functionalized_rng_ops_wrapper(helper, args, trace_joint)
109 create_functionalized_rng_ops_wrapper,