| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 65 maximum_iterations=None, argument
106 maximum_iterations)
110 if maximum_iterations is not None else None,
137 if maximum_iterations is None:
248 max_num_elements=maximum_iterations)
322 maximum_iterations = op.inputs[1]
356 util.unique_grad_fn_name(body_graph.name), op, maximum_iterations)
609 maximum_iterations): argument
644 args = [counter, maximum_iterations, total_iters] + list(grads)
652 maximum_iterations, while_op,
[all …]
|
| D | control_flow_ops.py | 1457 maximum_iterations=None, argument
1483 self._init_from_args(maximum_iterations, parallel_iterations, back_prop,
1488 def _init_from_args(self, maximum_iterations, parallel_iterations, back_prop, argument
1506 self._maximum_iterations = maximum_iterations
1578 def maximum_iterations(self): member in WhileContext
2372 maximum_iterations=None, argument
2540 maximum_iterations=maximum_iterations,
2554 maximum_iterations=None, argument
2735 maximum_iterations=maximum_iterations,
2744 if maximum_iterations is not None:
[all …]
|
| D | control_flow_state.py | 69 max_iter = while_ctxt.maximum_iterations
183 maximum_iterations=forward_ctxt.maximum_iterations,
197 maximum_iterations=forward_ctxt.maximum_iterations,
|
| D | functional_ops.py | 156 maximum_iterations=n)
352 maximum_iterations=n)
670 maximum_iterations=n)
|
| D | control_flow_ops_test.py | 484 def _testWhileContextHelper(self, maximum_iterations=None): argument
490 c, b, [i], maximum_iterations=maximum_iterations)
505 self._testWhileContextHelper(maximum_iterations=10)
1482 r = control_flow_ops.while_loop(c, b, [i, []], maximum_iterations=50)
1498 r = control_flow_ops.while_loop(c, b, [i], maximum_iterations=50)
1514 c, b, [i, []], return_same_structure=True, maximum_iterations=50)
1529 c, b, [i], return_same_structure=True, maximum_iterations=50)
|
| D | map_fn.py | 507 maximum_iterations=n)
|
| D | rnn.py | 893 maximum_iterations=time_steps,
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | directives.py | 53 maximum_iterations=UNSPECIFIED, argument
97 del maximum_iterations
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 149 maximum_iterations=iterations,
184 maximum_iterations=iterations,
195 maximum_iterations=iterations,
202 maximum_iterations=5,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils_test.cc | 486 input_arg { name: "maximum_iterations" type: DT_INT32 }
568 input: "maximum_iterations"
585 name: "maximum_iterations"
643 input: "maximum_iterations:output:0"
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 275 maximum_iterations=None, argument
286 maximum_iterations=maximum_iterations,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.autograph.experimental.pbtxt | 13 …argspec: "args=[\'parallel_iterations\', \'swap_memory\', \'maximum_iterations\', \'shape_invarian…
|
| D | tensorflow.pbtxt | 2529 …parallel_iterations\', \'back_prop\', \'swap_memory\', \'name\', \'maximum_iterations\', \'return_…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.autograph.experimental.pbtxt | 13 …argspec: "args=[\'parallel_iterations\', \'swap_memory\', \'maximum_iterations\', \'shape_invarian…
|
| D | tensorflow.pbtxt | 1165 …ants\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'maximum_iterations\', \'name\']…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | graph-input-func-arg-name-collision.pbtxt | 34 name: "while/maximum_iterations"
79 input: "while/maximum_iterations"
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | auto_control_deps_test.py | 224 lambda i: True, body, [0.0], maximum_iterations=1)
333 lambda i: True, body, [0.0], maximum_iterations=1)
414 lambda i: True, body, [0.0], maximum_iterations=1)
530 lambda i: True, body, [0.0], maximum_iterations=1)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | while_v2_test.py | 711 lambda _: True, lambda x: v * x, [1.0], maximum_iterations=5)
806 maximum_iterations=iterations,
817 maximum_iterations=iterations,
825 maximum_iterations=5,
847 maximum_iterations=5,
873 lambda _: True, lambda x: v * x, [1.0], maximum_iterations=5)
891 lambda _: True, lambda x: v * x, [1.0], maximum_iterations=5)
2027 maximum_iterations=2)
|
| D | control_flow_ops_py_test.py | 136 def isum(s, maximum_iterations=None): argument
141 c, b, [i, s], maximum_iterations=maximum_iterations)
1612 r = isum(s, maximum_iterations=3)
1619 lambda i: i < 3, lambda i: i + 1, [0], maximum_iterations=1)
1641 cond, body, (input1, input2), maximum_iterations=2)
1662 maximum_iterations=i)
1690 maximum_iterations=i)
1693 def create_while_loop(maximum_iterations=None): argument
1697 maximum_iterations=maximum_iterations)
1699 loop_no_xla = create_while_loop(maximum_iterations=5)
[all …]
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | control_flow.proto | 84 // Optional name of the maximum_iterations tensor.
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | known_issues.md | 30 *Workaround*: all compiled while loops need to either have `maximum_iterations`
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function_xla_jit_test.py | 232 maximum_iterations=10)[1]
336 maximum_iterations=10)[1]
|