Esempio n. 1
0
 def default_inputs(self):
     return DefaultInputs(
         axes=range(self.x.rank),
         gamma=None,
         beta=None,
         epsilon=1e-5,
     )
Esempio n. 2
0
 def default_inputs(self):
     return DefaultInputs(
         target_height=1,
         target_width=1,
         normalized_coordinates=False,
         spatial_scale=1.,
         box_coordinate_mode="CONRNERS_HEIGHT_FIRST",
         sampling_mode="DEFAULT",
     )
Esempio n. 3
0
 def default_inputs(self):
     num_spatial_dims = self.x.rank - 2
     return DefaultInputs(
         bias=None,
         strides=[1] * num_spatial_dims,
         pad_type="valid",
         pad=[0] * num_spatial_dims * 2,
         dilations=[1] * num_spatial_dims,
         groups=1,
     )
Esempio n. 4
0
 def default_inputs(self):
     return super().default_inputs() + \
         DefaultInputs(
             nbits=8,
             )
Esempio n. 5
0
 def default_inputs(self):
     return DefaultInputs(
         target_size_height=1,
         target_size_width=1,
         sampling_mode="DEFAULT",
     )
Esempio n. 6
0
 def default_inputs(self):
     return DefaultInputs(
         scale_factor_height=1,
         scale_factor_width=1,
         align_corners=True,
     )
Esempio n. 7
0
 def default_inputs(self):
     return DefaultInputs(
         scale_factor_height=1,
         scale_factor_width=1,
     )
Esempio n. 8
0
 def default_inputs(self):
     return DefaultInputs(
         gamma=None,
         beta=None,
         epsilon=1e-5,
     )
Esempio n. 9
0
 def default_inputs(self):
     return DefaultInputs(
         alpha=1e-4,
         beta=0.75,
         k=1.,
     )
Esempio n. 10
0
 def default_inputs(self):
     return DefaultInputs(epsilon=1e-6, )
Esempio n. 11
0
 def default_inputs(self):
     Dout = self.weight.shape[0]
     return DefaultInputs(bias=[0.] * Dout, )
Esempio n. 12
0
 def default_inputs(self):
     return DefaultInputs(
         transpose_x=False,
         transpose_y=False,
     )