Searched refs:custom_function_call_vmap_generate_rule (Results 1 – 1 of 1) sorted by relevance
300 return custom_function_call_vmap_generate_rule(357 def custom_function_call_vmap_generate_rule(interpreter, autograd_function, *operands): function