| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | cond_v2_test.py | 53 def _testCond(self, true_fn, false_fn, train_vals, feed_dict=None): argument
99 def true_fn(): function
113 def true_fn(): function
129 def true_fn(): function
166 def true_fn(): function
181 def true_fn(): function
196 def true_fn(): function
212 def true_fn(): function
228 def true_fn(): function
294 def true_fn(): function
[all …]
|
| H A D | control_flow_ops_py_test.py | 712 def true_fn(): function
753 def true_fn(): function
790 def true_fn(): function
827 def true_fn(): function
860 def true_fn(inputs): function
1128 def true_fn(): function
1153 def true_fn(): function
1173 def true_fn(): function
1327 def true_fn(): function
4728 def true_fn(): function
[all …]
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 269 def true_fn(x): function
281 def true_fn(x): function
293 def true_fn(x): function
335 def true_fn(x): function
379 def true_fn(x0, y0, z0): function
417 def true_fn(x): function
442 def true_fn(x, y, z): function
489 def true_fn(x): function
557 def true_fn(x): function
609 def true_fn(x): function
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | cond.py | 54 def __call__(self, pred, true_fn, false_fn, operands): argument
62 def cond(pred, true_fn, false_fn, operands): argument
142 def _validate_input(pred, true_fn, false_fn, operands): argument
180 def create_fw_bw_graph_branches(true_fn, false_fn, *operands): argument
221 def trace_cond(proxy_mode, func_overload, pred, true_fn, false_fn, operands): argument
344 def cond_op_dense(pred, true_fn, false_fn, operands): argument
386 def cond_autograd(pred, true_fn, false_fn, operands): argument
415 def inner(mode, pred, true_fn, false_fn, operands): argument
420 def cond_fake_tensor_mode(mode, pred, true_fn, false_fn, operands): argument
449 def cond_func(ctx, pred, true_fn, false_fn, inputs): argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_control_flow.py | 39 def true_fn(x, y): function
49 def true_fn(x0, y0, z0): function
113 def true_fn(x, y): function
127 def true_fn(x, y, z): function
140 def true_fn(x, y): function
155 def true_fn(x): function
165 def true_fn(x, y): function
239 def true_fn(x: torch.Tensor): function
316 def true_fn(x, y): function
400 def true_fn(x, y): function
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | auto_control_deps_test.py | 663 def true_fn(): function
685 def true_fn(): function
710 def true_fn(): function
744 def true_fn(): function
766 def true_fn(): function
787 def true_fn(): function
|
| H A D | convert_to_constants_test.py | 370 def true_fn(x): function
394 def true_fn(): function
735 def true_fn(x): function
760 def true_fn(): function
|
| H A D | smart_cond.py | 24 def smart_cond(pred, true_fn=None, false_fn=None, name=None): argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_export.py | 1710 def true_fn(val): function
1754 def true_fn(self, val): member in ExportTests.test_export_with_cond_branches_calling_methods.Module
1780 def true_fn(x): function
1793 def true_fn(x): function
1809 def true_fn(x, y): function
1834 def true_fn(x): function
1854 def true_fn(x): function
1864 def true_fn(x): function
1923 def true_fn(x): function
2099 def true_fn(self, x, y): member in ExportTests.test_export_cond_in_aten_symbolic.ConditionOp
[all …]
|
| H A D | test_higher_order_ops.py | 559 def _(pred, true_fn, false_fn, operands): argument
596 def _(mode, pred, true_fn, false_fn, operands): argument
1420 def true_fn(pred2, x, y): function
1465 def true_fn(x): function
1501 def true_fn(x): function
1579 def true_fn(): function
1622 def true_fn(): function
1674 def true_fn(x): function
1703 def true_fn(x): function
2304 def true_fn(pred2, x, y): function
[all …]
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_verifier.py | 63 def true_fn(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: function
82 def true_fn(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: function
|
| H A D | test_pass_infra.py | 51 def true_fn(x, y): function
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_tracer.py | 500 def true_fn(x): function
531 def true_fn(x): function
|
| /aosp_15_r20/external/pytorch/torch/_export/db/examples/ |
| H A D | cond_closed_over_variable.py | 12 def true_fn(val): function
|
| H A D | cond_branch_nonlocal_variables.py | 41 def true_fn(x, y, z): function
|
| H A D | cond_branch_nested_function.py | 22 def true_fn(x): function
|
| H A D | cond_operands.py | 21 def true_fn(x, y): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/ |
| H A D | conditional_expressions.py | 36 def true_fn(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/ |
| H A D | control_flow_util.py | 87 def smart_cond(pred, true_fn=None, false_fn=None, name=None): # pylint: disable=invalid-name argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_getitem.py | 461 def _if_ge_zero(value, true_fn, false_fn): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | control_flow_ops.py | 1055 def _eager_cond_implementation(pred, true_fn, false_fn, strict, name): argument
1106 true_fn=None, argument
1326 def cond_for_tf_v2(pred, true_fn=None, false_fn=None, name=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/layers/ |
| H A D | utils.py | 176 def smart_cond(pred, true_fn=None, false_fn=None, name=None): argument
|
| /aosp_15_r20/external/executorch/exir/backend/test/ |
| H A D | test_backends.py | 1018 def true_fn(x, y): function
1139 def true_fn(x, pred2): function
|
| H A D | test_backends_lifted.py | 1004 def true_fn(x, y): function
1127 def true_fn(x, pred2): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A D | lite_v2_test.py | 972 def true_fn(x): function
3123 def true_fn(x): function
3154 def true_fn(x): function
|