Searched refs:para_type (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | model_builder.py | 33 t.para_type = Tensor.PARA_INPUT
39 t.para_type = Tensor.PARA_OUTPUT
72 …def tensor(self, shape, dtype, data_format="DefaultFormat", name=None, para_type=Tensor.PARA_NONE): argument
78 return Tensor(name, shape, dtype, data_format, para_type=para_type)
147 shape, dtype, data_format, name=name, para_type=Tensor.PARA_INPUT)
152 shape, dtype, data_format, name=name, para_type=Tensor.PARA_OUTPUT)
160 inputs[1].para_type = Tensor.PARA_OUTPUT
|
| D | model.py | 309 def __init__(self, name, shape, dtype, data_format=DataFormat.DEFAULT, para_type=0): argument
314 self.para_type = para_type
434 if op.output.para_type == Tensor.PARA_OUTPUT or not op.output.to_ops:
|