Searched refs:_two_layer_model (Results 1 – 1 of 1) sorted by relevance
69 def _two_layer_model(x): function119 outputs = map_fn.map_fn(_two_layer_model, elems, dtype=dtypes.float32)279 output = _two_layer_model(x)309 conv = _two_layer_model(x)345 conv = _two_layer_model(x)380 conv = _two_layer_model(x)415 conv = _two_layer_model(x)445 conv = _two_layer_model(x)476 conv = _two_layer_model(x)507 conv = _two_layer_model(x)[all …]