Home
last modified time | relevance | path

Searched full:control_flow (Results 1 – 25 of 468) sorted by relevance

12345678910>>...19

/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/
H A Dcontrol_flow_test.py15 """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 …]
H A Dconditional_expressions.py18 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__.py37 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/tensorflow/tensorflow/python/autograph/converters/
H A Dcontrol_flow_test.py15 """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 …]
/aosp_15_r20/external/executorch/backends/apple/mps/test/
H A Dtest_mps_models.py8 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/pytorch/test/
H A Dtest_fx.py22 from functorch.experimental import control_flow
358 x = control_flow.cond(x[0] == 0, true, false, [x, y])
4288CONTROL_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 …]
/aosp_15_r20/external/executorch/test/end2end/
H A Dtest_end2end.py18 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 Dtest_control_flow.py8 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/executorch/exir/
H A Dcontrol_flow.py33 @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}",
H A DTARGETS95 name = "control_flow",
97 "control_flow.py",
158 ":control_flow",
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A DBUILD100 "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 Dtest_backends_nested.py35 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/
Deedb42bef8e672dbde62cfdb6848327ddb65fe3623 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/
D1d553c8e5c51377a0735f782de2195bf126ffdb223 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/
Ddd6487dc48b06a4193fe2df7f0db952c0066f63123 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/
Dcb874114bd53482187d014ae834fe3bb097637a923 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/
D4d4644fab55cfb07cffd073e1cd65ee933072abe23 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/
Dgenrule.sbox.textproto23 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/
Dgenrule.sbox.textproto23 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/tensorflow/tensorflow/python/autograph/g3doc/reference/
H A Doperators.md118 …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)
/aosp_15_r20/external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/khronos_mustpass/3.2.2.x/
H A Dgles2-gtf-main.txt254 GTF-GLES2.gtf.GL.control_flow.for_break_frag
255 GTF-GLES2.gtf.GL.control_flow.for_break_vert
256 GTF-GLES2.gtf.GL.control_flow.for_continue_frag
257 GTF-GLES2.gtf.GL.control_flow.for_continue_vert
258 GTF-GLES2.gtf.GL.control_flow.for_nested_break_frag
259 GTF-GLES2.gtf.GL.control_flow.for_nested_break_vert
260 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_frag
261 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_vert
262 GTF-GLES2.gtf.GL.control_flow.nested_if_else_frag
263 GTF-GLES2.gtf.GL.control_flow.nested_if_else_vert
/aosp_15_r20/external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/khronos_mustpass/3.2.3.x/
H A Dgles2-gtf-main.txt254 GTF-GLES2.gtf.GL.control_flow.for_break_frag
255 GTF-GLES2.gtf.GL.control_flow.for_break_vert
256 GTF-GLES2.gtf.GL.control_flow.for_continue_frag
257 GTF-GLES2.gtf.GL.control_flow.for_continue_vert
258 GTF-GLES2.gtf.GL.control_flow.for_nested_break_frag
259 GTF-GLES2.gtf.GL.control_flow.for_nested_break_vert
260 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_frag
261 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_vert
262 GTF-GLES2.gtf.GL.control_flow.nested_if_else_frag
263 GTF-GLES2.gtf.GL.control_flow.nested_if_else_vert
/aosp_15_r20/external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/khronos_mustpass/3.2.3.x/src/
H A Dgles2-gtf-main.txt254 GTF-GLES2.gtf.GL.control_flow.for_break_frag
255 GTF-GLES2.gtf.GL.control_flow.for_break_vert
256 GTF-GLES2.gtf.GL.control_flow.for_continue_frag
257 GTF-GLES2.gtf.GL.control_flow.for_continue_vert
258 GTF-GLES2.gtf.GL.control_flow.for_nested_break_frag
259 GTF-GLES2.gtf.GL.control_flow.for_nested_break_vert
260 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_frag
261 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_vert
262 GTF-GLES2.gtf.GL.control_flow.nested_if_else_frag
263 GTF-GLES2.gtf.GL.control_flow.nested_if_else_vert
/aosp_15_r20/external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/khronos_mustpass/3.2.2.x/src/
H A Dgles2-gtf-main.txt254 GTF-GLES2.gtf.GL.control_flow.for_break_frag
255 GTF-GLES2.gtf.GL.control_flow.for_break_vert
256 GTF-GLES2.gtf.GL.control_flow.for_continue_frag
257 GTF-GLES2.gtf.GL.control_flow.for_continue_vert
258 GTF-GLES2.gtf.GL.control_flow.for_nested_break_frag
259 GTF-GLES2.gtf.GL.control_flow.for_nested_break_vert
260 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_frag
261 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_vert
262 GTF-GLES2.gtf.GL.control_flow.nested_if_else_frag
263 GTF-GLES2.gtf.GL.control_flow.nested_if_else_vert
/aosp_15_r20/external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/khronos_mustpass/3.2.4.x/src/
H A Dgles2-gtf-main.txt254 GTF-GLES2.gtf.GL.control_flow.for_break_frag
255 GTF-GLES2.gtf.GL.control_flow.for_break_vert
256 GTF-GLES2.gtf.GL.control_flow.for_continue_frag
257 GTF-GLES2.gtf.GL.control_flow.for_continue_vert
258 GTF-GLES2.gtf.GL.control_flow.for_nested_break_frag
259 GTF-GLES2.gtf.GL.control_flow.for_nested_break_vert
260 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_frag
261 GTF-GLES2.gtf.GL.control_flow.for_nested_continue_vert
262 GTF-GLES2.gtf.GL.control_flow.nested_if_else_frag
263 GTF-GLES2.gtf.GL.control_flow.nested_if_else_vert

12345678910>>...19