Searched refs:_model_with_branch (Results 1 – 1 of 1) sorted by relevance
97 def _model_with_branch(x): function134 outputs = map_fn.map_fn(_model_with_branch, elems, dtype=dtypes.float32)