!40333 remove unused lines
Merge pull request !40333 from 李林杰/0811_remove_unused_lines_master
This commit is contained in:
commit
79e655eefd
|
@ -1247,7 +1247,6 @@ class MaxPool3DGradWithArgmax(Primitive):
|
|||
|
||||
@prim_attr_register
|
||||
def __init__(self, ksize, strides, pads, dilation=(1, 1, 1), ceil_mode=False, data_format="NCDHW"):
|
||||
self.add_prim_attr("cust_aicpu", self.name)
|
||||
self.init_prim_io_names(inputs=['x', 'grads', 'argmax'], outputs=['y'])
|
||||
validator.check_value_type('ceil_mode', ceil_mode, bool, self.name)
|
||||
validator.check_value_type('data_format', data_format, str, self.name)
|
||||
|
|
|
@ -612,7 +612,6 @@ class Col2Im(Primitive):
|
|||
@prim_attr_register
|
||||
def __init__(self, kernel_size, dilation=1, padding=0, stride=1):
|
||||
"""Initialize Col2Im."""
|
||||
self.add_prim_attr("cust_aicpu", self.name)
|
||||
self.init_prim_io_names(inputs=['x', 'output_size'], outputs=['y'])
|
||||
validator.check_value_type('kernel_size', kernel_size, [int, list, tuple], self.name)
|
||||
validator.check_value_type('dilation', dilation, [int, list, tuple], self.name)
|
||||
|
|
|
@ -2519,7 +2519,6 @@ class MaxPool3DWithArgmax(Primitive):
|
|||
def __init__(self, ksize, strides, pads, dilation=(1, 1, 1), ceil_mode=False,
|
||||
data_format="NCDHW", argmax_type=mstype.int64):
|
||||
"""Initialize MaxPool3DWithArgmax."""
|
||||
self.add_prim_attr("cust_aicpu", self.name)
|
||||
self.init_prim_io_names(inputs=['x'], outputs=['y', 'argmax'])
|
||||
validator.check_value_type('ceil_mode', ceil_mode, bool, self.name)
|
||||
validator.check_value_type('data_format', data_format, str, self.name)
|
||||
|
|
Loading…
Reference in New Issue