| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/ |
| H A D | control_flow_test.py | 27 from tensorflow.python.autograph.operators import control_flow
72 control_flow.for_stmt(
92 control_flow.for_stmt(
114 control_flow.for_stmt(
136 control_flow.for_stmt(
159 control_flow.for_stmt(
182 control_flow.for_stmt(
211 control_flow.for_stmt(
233 control_flow.for_stmt(
260 control_flow.for_stmt(
[all …]
|
| H A D | conditional_expressions.py | 18 from tensorflow.python.autograph.operators import control_flow
39 control_flow.verify_single_cond_var(expr_repr, true_val[0], false_val[0])
45 control_flow.verify_single_cond_var(expr_repr, true_val[0], false_val[0])
|
| H A D | __init__.py | 37 from tensorflow.python.autograph.operators.control_flow import for_stmt
38 from tensorflow.python.autograph.operators.control_flow import if_stmt
39 from tensorflow.python.autograph.operators.control_flow import while_stmt
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps_models.py | 8 import executorch.exir.control_flow as control_flow namespace
10 from functorch.experimental.control_flow import cond
230 @control_flow.tracing_context(
236 @control_flow.tracing_context(
241 @control_flow.tracing_context(
247 @control_flow.tracing_context(
260 y, _ = control_flow.while_loop(
276 @control_flow.tracing_context(
280 @control_flow.tracing_context(
289 @control_flow.tracing_context(
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/converters/ |
| H A D | control_flow_test.py | 23 from tensorflow.python.autograph.converters import control_flow
51 tr = self.transform(f, control_flow)
116 tr = self.transform(f, control_flow)
200 tr = self.transform(f, control_flow)
220 tr = self.transform(f, control_flow)
239 tr = self.transform(f, control_flow)
256 tr = self.transform(f, control_flow)
333 tr = self.transform(f, control_flow)
359 tr = self.transform(f, control_flow)
413 tr = self.transform(f, control_flow)
[all …]
|
| /aosp_15_r20/external/executorch/test/end2end/ |
| H A D | test_end2end.py | 18 import executorch.exir.control_flow as control_flow namespace
50 from functorch.experimental.control_flow import cond
282 @control_flow.tracing_context(
288 @control_flow.tracing_context(
293 @control_flow.tracing_context(
299 @control_flow.tracing_context(
312 y, _ = control_flow.while_loop(
328 @control_flow.tracing_context(
332 @control_flow.tracing_context(
341 @control_flow.tracing_context(
[all …]
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 8 from functorch.experimental import control_flow
9 from functorch.experimental.control_flow import cond, UnsupportedAliasMutationException
70 from functorch.experimental.control_flow import _stack_pytree, _unstack_pytree
1044 res = control_flow.map(f, xs, y)
1069 _ = control_flow.map(f, (3, torch.ones(2)), torch.ones(2))
1074 _ = control_flow.map(
1083 _ = control_flow.map(
1102 _ = control_flow.map(f, x, y)
1107 out = control_flow.map(f1, x, y)
1110 _ = control_flow.map(f2, x, y)
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | BUILD | 100 "control_flow.h",
125 "control_flow.cc",
166 "control_flow.cc",
167 "control_flow.h",
|
| /aosp_15_r20/external/executorch/exir/backend/test/ |
| H A D | test_backends_nested.py | 35 from functorch.experimental import control_flow
54 z = control_flow.cond(pred2, true_nested, false_nested, [x])
62 y = control_flow.cond(pred1, true_fn, false_fn, [x, pred2])
189 if node.target is not control_flow.cond:
|
| /aosp_15_r20/out/soong/raw-aosp_shiba/ee/ |
| D | eedb42bef8e672dbde62cfdb6848327ddb65fe36 | 23 from: "art/runtime/interpreter/mterp/x86ng/control_flow.S"
24 to: "art/runtime/interpreter/mterp/x86ng/control_flow.S"
47 …e/interpreter/mterp/x86ng/array.S art/runtime/interpreter/mterp/x86ng/control_flow.S art/runtime/i…
|
| /aosp_15_r20/out/soong/raw-aosp_shiba/1d/ |
| D | 1d553c8e5c51377a0735f782de2195bf126ffdb2 | 23 from: "art/runtime/interpreter/mterp/armng/control_flow.S"
24 to: "art/runtime/interpreter/mterp/armng/control_flow.S"
47 …e/interpreter/mterp/armng/array.S art/runtime/interpreter/mterp/armng/control_flow.S art/runtime/i…
|
| /aosp_15_r20/out/soong/raw-aosp_shiba/dd/ |
| D | dd6487dc48b06a4193fe2df7f0db952c0066f631 | 23 from: "art/runtime/interpreter/mterp/riscv64/control_flow.S"
24 to: "art/runtime/interpreter/mterp/riscv64/control_flow.S"
47 …terpreter/mterp/riscv64/array.S art/runtime/interpreter/mterp/riscv64/control_flow.S art/runtime/i…
|
| /aosp_15_r20/out/soong/raw-aosp_shiba/cb/ |
| D | cb874114bd53482187d014ae834fe3bb097637a9 | 23 from: "art/runtime/interpreter/mterp/arm64ng/control_flow.S"
24 to: "art/runtime/interpreter/mterp/arm64ng/control_flow.S"
47 …terpreter/mterp/arm64ng/array.S art/runtime/interpreter/mterp/arm64ng/control_flow.S art/runtime/i…
|
| /aosp_15_r20/out/soong/raw-aosp_shiba/4d/ |
| D | 4d4644fab55cfb07cffd073e1cd65ee933072abe | 23 from: "art/runtime/interpreter/mterp/x86_64ng/control_flow.S"
24 to: "art/runtime/interpreter/mterp/x86_64ng/control_flow.S"
47 …rpreter/mterp/x86_64ng/array.S art/runtime/interpreter/mterp/x86_64ng/control_flow.S art/runtime/i…
|
| /aosp_15_r20/out/soong/.intermediates/art/runtime/libart_mterp.x86_64ng/ |
| D | genrule.sbox.textproto | 23 from: "art/runtime/interpreter/mterp/x86_64ng/control_flow.S"
24 to: "art/runtime/interpreter/mterp/x86_64ng/control_flow.S"
47 …rpreter/mterp/x86_64ng/array.S art/runtime/interpreter/mterp/x86_64ng/control_flow.S art/runtime/i…
|
| /aosp_15_r20/out/soong/.intermediates/art/runtime/libart_mterp.arm64ng/ |
| D | genrule.sbox.textproto | 23 from: "art/runtime/interpreter/mterp/arm64ng/control_flow.S"
24 to: "art/runtime/interpreter/mterp/arm64ng/control_flow.S"
47 …terpreter/mterp/arm64ng/array.S art/runtime/interpreter/mterp/arm64ng/control_flow.S art/runtime/i…
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | TARGETS | 95 name = "control_flow",
97 "control_flow.py",
158 ":control_flow",
|
| /aosp_15_r20/external/executorch/devtools/bundled_program/test/ |
| H A D | test_end2end.py | 17 import executorch.exir.control_flow as control_flow namespace
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_verifier.py | 5 from functorch.experimental import control_flow
69 return control_flow.cond(x.sum() > 2, true_fn, false_fn, [x, y])
88 return control_flow.cond(x.sum() > 2, true_fn, false_fn, [x, y])
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_higher_order_ops.py | 9 import functorch.experimental.control_flow as control_flow namespace
1179 return control_flow.map(inner2, x, y)
1181 return control_flow.map(inner, xs, y)
1212 return control_flow.map(lambda x: (x.sin(), x.sin()), x)
1248 return control_flow.map(inner_f, x)
1282 return control_flow.map(lambda x: x.sin(), x=x)
1296 return control_flow.map(inner, x, y.size(0))
1330 return control_flow.map(inner, x, y.size(0))
1383 control_flow.map, backend=backend, fullgraph=True
1430 return control_flow.cond(pred2, true_fn2, false_fn2, [x, y])
[all …]
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.80.1/lib/rustlib/src/rust/library/core/src/ops/ |
| H A D | mod.rs | 144 mod control_flow; module
220 pub use self::control_flow::ControlFlow;
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.81.0.u1/lib/rustlib/src/rust/library/core/src/ops/ |
| H A D | mod.rs | 144 mod control_flow; module
220 pub use self::control_flow::ControlFlow;
|
| /aosp_15_r20/prebuilts/rust/linux-musl-x86/1.81.0/lib/rustlib/src/rust/library/core/src/ops/ |
| H A D | mod.rs | 144 mod control_flow; module
220 pub use self::control_flow::ControlFlow;
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.81.0/lib/rustlib/src/rust/library/core/src/ops/ |
| H A D | mod.rs | 144 mod control_flow; module
220 pub use self::control_flow::ControlFlow;
|
| /aosp_15_r20/prebuilts/rust/linux-musl-x86/1.80.1/lib/rustlib/src/rust/library/core/src/ops/ |
| H A D | mod.rs | 144 mod control_flow; module
220 pub use self::control_flow::ControlFlow;
|