Home
last modified time | relevance | path

Searched defs:outer_body_fn (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_higher_order_ops.py2521 def outer_body_fn(x, y): function
2582 def outer_body_fn(x, y): function
2601 def outer_body_fn(x, y): function
2620 def outer_body_fn(x): function
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dsaver_test.py2336 def _testWhileLoopAndGradientSerDes(self, outer_body_fn): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_control_flow_test.cc1153 NameAttrList outer_cond_fn, outer_body_fn; in CheckOuterNodesFunctionalized() local
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_control_flow.py144 def outer_body_fn(out_iter, it, y): function
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_export.py7421 def outer_body_fn(x, y): function