Home
last modified time | relevance | path

Searched refs:extra_test (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/operators/
Dcontrol_flow_deprecated_py2.py280 extra_test, argument
330 return _tf_range_for_stmt(iter_, extra_test, body, get_state, set_state,
334 return _known_len_tf_for_stmt(iter_, extra_test, body, get_state,
339 return _tf_dataset_for_stmt(iter_, extra_test, body, get_state, set_state,
344 return _tf_iterator_for_stmt(iter_, extra_test, body, get_state, set_state,
349 return _tf_ragged_for_stmt(iter_, extra_test, body, get_state, set_state,
359 return _tf_distributed_dataset_for_stmt(iter_, extra_test, body, init_vars)
361 return _py_for_stmt(iter_, extra_test, body, get_state, set_state, init_vars)
364 def _py_for_stmt(iter_, extra_test, body, get_state, set_state, init_vars): argument
369 if extra_test is not None:
[all …]
Dcontrol_flow.py369 def for_stmt(iter_, extra_test, body, get_state, set_state, symbol_names, opts): argument
412 _tf_range_for_stmt(iter_, extra_test, body, get_state, set_state,
415 _tf_ragged_for_stmt(iter_, extra_test, body, get_state, set_state,
419 iter_, extra_test, body, get_state, set_state, symbol_names, opts)
423 iter_, extra_test, body, get_state, set_state, symbol_names, opts)
427 iter_, extra_test, body, get_state, set_state, symbol_names, opts)
431 iter_, extra_test, body, get_state, set_state, symbol_names, opts)
435 iter_, extra_test, body, get_state, set_state, symbol_names, opts)
440 iter_, extra_test, body, get_state, set_state, symbol_names, opts)
443 _py_for_stmt(iter_, extra_test, body, None, None)
[all …]
Dcontrol_flow_test.py78 extra_test=lambda: True,
98 extra_test=lambda: True,
120 extra_test=lambda: True,
142 extra_test=lambda: True,
165 extra_test=lambda: True,
188 extra_test=lambda: True,
217 extra_test=lambda: state.field_1 < 6,
238 extra_test=lambda: True,
265 extra_test=lambda: c == 0, # Break after first iteration
296 extra_test=lambda: False, # Break before loop
[all …]
/external/tensorflow/tensorflow/python/autograph/converters/
Dreturn_statements.py304 extra_test = anno.getanno(node, anno.Basic.EXTRA_LOOP_TEST, default=None)
305 if extra_test is not None:
306 extra_test = templates.replace_as_expression(
308 extra_test=extra_test,
311 extra_test = templates.replace_as_expression(
314 anno.setanno(node, anno.Basic.EXTRA_LOOP_TEST, extra_test)
Dbreak_statements.py154 extra_test = templates.replace_as_expression(
177 anno.setanno(new_for_node, anno.Basic.EXTRA_LOOP_TEST, extra_test)
Dcontrol_flow.py336 extra_test = anno.getanno(node, anno.Basic.EXTRA_LOOP_TEST)
346 extra_test_expr=extra_test,
Dcontrol_flow_deprecated_py2.py507 extra_test = anno.getanno(node, anno.Basic.EXTRA_LOOP_TEST)
518 extra_test_expr=extra_test)
/external/tensorflow/tensorflow/python/autograph/g3doc/reference/
Dcontrol_flow.md384 corresponding operator accepts an `extra_test` argument which is similar to
418 def extra_test(break_):
421 break_, = ag__.for_stmt(range(10), extra_test, ..., (break_,))