Searched refs:_two_layer_model (Results 1 – 1 of 1) sorted by relevance
73 def _two_layer_model(x): function123 outputs = map_fn.map_fn(_two_layer_model, elems, dtype=dtypes.float32)262 output = _two_layer_model(x)292 conv = _two_layer_model(x)328 conv = _two_layer_model(x)363 conv = _two_layer_model(x)398 conv = _two_layer_model(x)428 conv = _two_layer_model(x)459 conv = _two_layer_model(x)490 conv = _two_layer_model(x)[all …]