!6863 update graphengine & adapt to r1.0

Merge pull request !6863 from HW_KK/r1.0
pull/6863/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit e266a0faa6

@ -1 +1 @@
Subproject commit 63a27119bf705511946f690c55c722c356f4b397
Subproject commit 86596e18efba333aed421fcc7aa453aa57de001a

@ -26,9 +26,8 @@ tile_op_info = TBERegOp("Tile") \
.attr("multiples", "optional", "listInt", "all")\
.input(0, "x1", False, "required", "all") \
.output(0, "y", False, "required", "all") \
.dtype_format(DataType.I32_Default, DataType.I32_Default) \
.dtype_format(DataType.I16_Default, DataType.I16_Default) \
.dtype_format(DataType.F32_Default, DataType.F32_Default) \
.op_pattern("dynamicFormat") \
.dtype_format(DataType.None_None, DataType.None_None) \
.get_op_info()

@ -548,8 +548,6 @@ class Model:
>>> model.train(2, dataset)
"""
check_bool(dataset_sink_mode)
if sink_size == -1:
sink_size = train_dataset.get_dataset_size()
check_int(sink_size)
if sink_size < -1 or sink_size == 0:
raise ValueError("The sink_size must be -1 or positive, but got sink_size {}.".format(sink_size))

@ -21,7 +21,7 @@ from mindspore import Tensor
class MindData:
""" Stub for MindData """
def __init__(self, size=1, batch_size=None, repeat_count=1,
def __init__(self, size=None, batch_size=None, repeat_count=1,
np_types=None, output_shapes=None, input_indexs=()):
self._size = size
self._batch_size = batch_size

@ -113,7 +113,7 @@ class TrainOneStepWithLossScaleCell(nn.Cell):
class DatasetLenet(MindData):
def __init__(self, predict, label, length=3):
super(DatasetLenet, self).__init__(size=length)
super(DatasetLenet, self).__init__()
self.predict = predict
self.label = label
self.index = 0

Loading…
Cancel
Save