def create_outer_variable_scope(self, device_num): self._custom_getter = variable_mgr_util.StagedVariableGetter( device_num, self.benchmark_cnn.raw_devices, self.benchmark_cnn.cpu_device, self) return tf.variable_scope('v', reuse=bool(device_num), custom_getter=self._custom_getter)
def create_outer_variable_scope(self, device_num): self._custom_getter = variable_mgr_util.StagedVariableGetter( device_num, self.benchmark_cnn.raw_devices, None, self) return tf.variable_scope('v', reuse=bool(device_num) or self._reuse_vars, custom_getter=self._custom_getter, use_resource=self.use_resource_vars)