Searched refs:_model_with_vec_and_4d (Results 1 – 1 of 1) sorted by relevance
103 def _model_with_vec_and_4d(x): function141 outputs = map_fn.map_fn(_model_with_vec_and_4d, elems, dtype=dtypes.float32)