Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py97 def _model_with_branch(x): function
134 outputs = map_fn.map_fn(_model_with_branch, elems, dtype=dtypes.float32)