| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_test.py | 15 """Tests for control_flow module."""
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(
[all …]
|
| 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])
|
| 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
|
| /external/executorch/backends/apple/mps/test/ |
| 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 …]
|
| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | control_flow_test.py | 15 """Tests for control_flow module."""
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)
[all …]
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/png-0.17.14/examples/ |
| D | show.rs | 71 event_loop.run(move |event, _, control_flow| match event { in main_loop()
75 } => exit(control_flow), in main_loop()
89 Some(VirtualKeyCode::Escape) => exit(control_flow), in main_loop()
100 exit(control_flow); in main_loop()
104 None => exit(control_flow), in main_loop()
123 fn exit(control_flow: &mut ControlFlow) { in exit()
124 *control_flow = ControlFlow::Exit; in exit()
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/png-0.17.14/examples/ |
| D | show.rs | 71 event_loop.run(move |event, _, control_flow| match event { in main_loop()
75 } => exit(control_flow), in main_loop()
89 Some(VirtualKeyCode::Escape) => exit(control_flow), in main_loop()
100 exit(control_flow); in main_loop()
104 None => exit(control_flow), in main_loop()
123 fn exit(control_flow: &mut ControlFlow) { in exit()
124 *control_flow = ControlFlow::Exit; in exit()
|
| /external/executorch/test/end2end/ |
| 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 …]
|
| /external/executorch/exir/ |
| D | control_flow.py | 33 @control_flow.tracing_context(inputs)
38 @control_flow.tracing_context(inputs)
43 ret = control_flow.cond(pred, branch_true, branch_false, args)
142 … f"control_flow.while_loop() expects all inputs values to be tensors, actual inputs: {init_val}",
154 … f"control_flow.while_loop() expects all returned values to be tensors, actual outputs: {val}",
|
| D | TARGETS | 95 name = "control_flow",
97 "control_flow.py",
158 ":control_flow",
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | BUILD | 100 "control_flow.h",
125 "control_flow.cc",
166 "control_flow.cc",
167 "control_flow.h",
|
| /external/pytorch/test/ |
| D | test_fx.py | 22 from functorch.experimental import control_flow
358 x = control_flow.cond(x[0] == 0, true, false, [x, y])
4288 …CONTROL_FLOW = (TraceError, r"symbolically traced variables cannot be used as inputs to control fl… variable in TestFunctionalTracing
4335 "instance_norm": CONTROL_FLOW,
4361 "affine_grid": CONTROL_FLOW,
4362 "alpha_dropout": CONTROL_FLOW,
4363 "batch_norm": CONTROL_FLOW,
4364 "binary_cross_entropy": CONTROL_FLOW,
4365 "binary_cross_entropy_with_logits": CONTROL_FLOW,
4366 "celu": CONTROL_FLOW,
[all …]
|
| /external/executorch/exir/backend/test/ |
| 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:
|
| /external/executorch/devtools/bundled_program/test/ |
| D | test_end2end.py | 17 import executorch.exir.control_flow as control_flow namespace
|
| /external/tensorflow/tensorflow/python/autograph/g3doc/reference/ |
| D | operators.md | 118 …lob/bacd16a95d5a6f3d5081e3d56c515671c784d289/tensorflow/python/autograph/operators/control_flow.py)
137 …acd16a95d5a6f3d5081e3d56c515671c784d289/tensorflow/python/autograph/operators/control_flow.py#L369)
222 …cd16a95d5a6f3d5081e3d56c515671c784d289/tensorflow/python/autograph/operators/control_flow.py#L1125)
273 …acd16a95d5a6f3d5081e3d56c515671c784d289/tensorflow/python/autograph/operators/control_flow.py#L811)
|
| D | index.md | 11 * [Control flow](control_flow.md)
|
| /external/pytorch/test/export/ |
| 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])
|
| D | test_pass_infra.py | 6 from functorch.experimental import control_flow
63 ret = control_flow.cond(pred, true_fn, false_fn, [x, y])
|
| /external/pytorch/test/functorch/ |
| 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 …]
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | BUILD | 167 "control_flow.proto",
198 "control_flow.proto",
|
| /external/mesa3d/src/nouveau/compiler/nak/ |
| D | opt_jump_thread.rs | 104 if let Some(control_flow) = block.branch() { in rewrite_cfg()
105 match &control_flow.op { in rewrite_cfg()
|
| /external/pytorch/torch/_export/db/examples/ |
| D | dynamic_shape_map.py | 4 from functorch.experimental.control_flow import map
|
| D | cond_closed_over_variable.py | 4 from functorch.experimental.control_flow import cond
|
| D | cond_predicate.py | 4 from functorch.experimental.control_flow import cond
|
| D | cond_operands.py | 5 from functorch.experimental.control_flow import cond
|