| /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
102 def false_fn(): function
115 def false_fn(): function
133 def false_fn(): function
169 def false_fn(): function
184 def false_fn(): function
199 def false_fn(): function
215 def false_fn(): function
231 def false_fn(): function
302 def false_fn(): function
[all …]
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 272 def false_fn(x): function
284 def false_fn(x): function
296 def false_fn(x): function
338 def false_fn(x): function
396 def false_fn(x0, y0, z0): function
422 def false_fn(x): function
445 def false_fn(x, y, z): function
492 def false_fn(X): function
560 def false_fn(x): function
612 def false_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 | 42 def false_fn(x, y): function
66 def false_fn(x0, y0, z0): function
118 def false_fn(x, y): function
130 def false_fn(x, y, z): function
143 def false_fn(x, y): function
158 def false_fn(x): function
168 def false_fn(x, y): function
242 def false_fn(x: torch.Tensor): function
319 def false_fn(x, y): function
404 def false_fn(x, y): function
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | auto_control_deps_test.py | 667 def false_fn(): function
689 def false_fn(): function
714 def false_fn(): function
747 def false_fn(): function
769 def false_fn(): function
790 def false_fn(): function
|
| H A D | convert_to_constants_test.py | 373 def false_fn(x): function
397 def false_fn(): function
738 def false_fn(x): function
763 def false_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 | 1713 def false_fn(val): function
1757 def false_fn(self, val): member in ExportTests.test_export_with_cond_branches_calling_methods.Module
1783 def false_fn(x): function
1796 def false_fn(x): function
1812 def false_fn(x, y): function
1837 def false_fn(x): function
1857 def false_fn(x): function
1867 def false_fn(x): function
1926 def false_fn(x): function
2102 def false_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
1423 def false_fn(pred2, x, y): function
1469 def false_fn(x): function
1504 def false_fn(x): function
1582 def false_fn(): function
1625 def false_fn(): function
1680 def false_fn(x): function
1706 def false_fn(x): function
2307 def false_fn(pred2, x, y): function
[all …]
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_verifier.py | 66 def false_fn(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: function
85 def false_fn(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: function
|
| H A D | test_pass_infra.py | 57 def false_fn(x, y): function
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_tracer.py | 503 def false_fn(x): function
534 def false_fn(x): function
|
| /aosp_15_r20/external/pytorch/torch/_export/db/examples/ |
| H A D | cond_closed_over_variable.py | 15 def false_fn(val): function
|
| H A D | cond_branch_nonlocal_variables.py | 44 def false_fn(x, y, z): function
|
| H A D | cond_branch_nested_function.py | 28 def false_fn(x): function
|
| H A D | cond_operands.py | 24 def false_fn(x, y): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/ |
| H A D | conditional_expressions.py | 42 def false_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
1107 false_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 | 1024 def false_fn(x, y): function
1143 def false_fn(x, _): function
|
| H A D | test_backends_lifted.py | 1010 def false_fn(x, y): function
1131 def false_fn(x, _): function
|
| H A D | test_backends_nested.py | 57 def false_fn(x, _pred2): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A D | lite_v2_test.py | 975 def false_fn(x): function
3126 def false_fn(x): function
3157 def false_fn(x): function
|