Searched refs:switch_layer (Results 1 – 3 of 3) sorted by relevance
261 return F.switch_layer(index, layers), _bprop_switch_layer
238 switch_layer = Primitive('switch_layer') variable
404 ret = F.switch_layer(index, self.layers)(x) * self.z3