Home
last modified time | relevance | path

Searched refs:maximum_iterations (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Ddecoder.py158 maximum_iterations=None, argument
164 self.maximum_iterations = maximum_iterations
175 maximum_iterations=self.maximum_iterations,
271 maximum_iterations=None, argument
327 if maximum_iterations is not None:
328 maximum_iterations = ops.convert_to_tensor(
329 maximum_iterations, dtype=dtypes.int32, name="maximum_iterations")
330 if maximum_iterations.get_shape().ndims != 0:
346 if is_xla and maximum_iterations is None:
348 if maximum_iterations is not None:
[all …]
Dbeam_search_decoder.py935 maximum_iterations=self.maximum_iterations,
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Ddecoder_v2_test.py38 def _testDecodeRNN(self, time_major, maximum_iterations=None): argument
61 maximum_iterations=maximum_iterations)
90 if maximum_iterations is not None:
91 time_steps = min(max_out, maximum_iterations)
92 expected_length = [min(x, maximum_iterations) for x in expected_length]
93 if context.executing_eagerly() and maximum_iterations != 0:
110 self._testDecodeRNN(time_major=True, maximum_iterations=0)
113 self._testDecodeRNN(time_major=True, maximum_iterations=1)
Ddecoder_test.py37 def _testDynamicDecodeRNN(self, time_major, maximum_iterations=None): argument
64 maximum_iterations=maximum_iterations))
95 if maximum_iterations is not None:
96 time_steps = min(max_out, maximum_iterations)
97 expected_length = [min(x, maximum_iterations) for x in expected_length]
114 self._testDynamicDecodeRNN(time_major=True, maximum_iterations=0)
117 self._testDynamicDecodeRNN(time_major=True, maximum_iterations=1)
Dbeam_search_decoder_test.py539 bsd, output_time_major=time_major, maximum_iterations=max_out))
647 maximum_iterations=max_out)
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py61 maximum_iterations=None, argument
89 maximum_iterations)
93 if maximum_iterations is not None else None,
112 if maximum_iterations is None:
259 maximum_iterations = op.inputs[1]
274 util.unique_grad_fn_name(body_graph.name), op, maximum_iterations)
424 maximum_iterations): argument
449 args = [counter, maximum_iterations, total_iters] + list(grads)
457 maximum_iterations))
509 maximum_iterations = args[1]
[all …]
Dcontrol_flow_ops.py608 max_iter = while_ctxt.maximum_iterations
722 maximum_iterations=forward_ctxt.maximum_iterations,
736 maximum_iterations=forward_ctxt.maximum_iterations,
2166 maximum_iterations=None, argument
2192 self._init_from_args(maximum_iterations, parallel_iterations, back_prop,
2197 def _init_from_args(self, maximum_iterations, parallel_iterations, back_prop, argument
2215 self._maximum_iterations = maximum_iterations
2287 def maximum_iterations(self): member in WhileContext
3049 maximum_iterations=None, argument
3215 maximum_iterations=maximum_iterations,
[all …]
Dfunctional_ops.py145 maximum_iterations=n)
256 maximum_iterations=n)
475 maximum_iterations=n)
Dmap_fn.py268 maximum_iterations=n)
Dcontrol_flow_ops_test.py438 def _testWhileContextHelper(self, maximum_iterations=None): argument
444 c, b, [i], maximum_iterations=maximum_iterations)
459 self._testWhileContextHelper(maximum_iterations=10)
Drnn.py878 maximum_iterations=time_steps,
/external/tensorflow/tensorflow/python/autograph/lang/
Ddirectives.py52 maximum_iterations=UNSPECIFIED): argument
68 del maximum_iterations
/external/tensorflow/tensorflow/compiler/tests/
Dwhile_test.py148 maximum_iterations=iterations,
183 maximum_iterations=iterations,
194 maximum_iterations=iterations,
201 maximum_iterations=5,
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils_test.cc474 input_arg { name: "maximum_iterations" type: DT_INT32 }
556 input: "maximum_iterations"
573 name: "maximum_iterations"
631 input: "maximum_iterations:output:0"
/external/tensorflow/tensorflow/contrib/constrained_optimization/python/
Dswap_regret_optimizer.py67 maximum_iterations=100): argument
89 if maximum_iterations <= 0:
94 not_done = (iteration < maximum_iterations)
/external/tensorflow/tensorflow/core/protobuf/
Dcontrol_flow.proto83 // Optional name of the maximum_iterations tensor.
/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow.py118 opts=dict(maximum_iterations=n))
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py129 def isum(s, maximum_iterations=None): argument
134 c, b, [i, s], maximum_iterations=maximum_iterations)
1364 r = isum(s, maximum_iterations=3)
1371 lambda i: i < 3, lambda i: i + 1, [0], maximum_iterations=1)
1383 maximum_iterations=i)
1411 maximum_iterations=i)
1414 def create_while_loop(maximum_iterations=None): argument
1418 maximum_iterations=maximum_iterations)
1420 loop_no_xla = create_while_loop(maximum_iterations=5)
1428 loop_with_maxiter = create_while_loop(maximum_iterations=2)
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_map_ops.py303 maximum_iterations=n)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dmasked_autoregressive.py289 maximum_iterations=static_event_size)
/external/tensorflow/tensorflow/python/autograph/
DLIMITATIONS.md14 …over TF datasets. Converts to `tf.while_loop`, with an additional `maximum_iterations` hint, if th…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt1073 …ants\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'maximum_iterations\', \'name\']…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2421 …parallel_iterations\', \'back_prop\', \'swap_memory\', \'name\', \'maximum_iterations\', \'return_…