Beispiel #1
0
 def _weight_variable(self,
                      shape,
                      init_method=None,
                      dtype=tf.float32,
                      init_param=None,
                      wd=None,
                      name=None,
                      trainable=True,
                      seed=0):
     return weight_variable_cpu(shape,
                                init_method=init_method,
                                dtype=dtype,
                                init_param=init_param,
                                wd=wd,
                                name=name,
                                trainable=trainable,
                                seed=seed)
 def _weight_variable(self,
                      shape,
                      init_method=None,
                      dtype=tf.float32,
                      init_param=None,
                      wd=None,
                      name=None,
                      trainable=True,
                      seed=0):
     """Wrapper to declare variables. Default on CPU."""
     return weight_variable_cpu(shape,
                                init_method=init_method,
                                dtype=dtype,
                                init_param=init_param,
                                wd=wd,
                                name=name,
                                trainable=trainable,
                                seed=seed)