Home
last modified time | relevance | path

Searched refs:relation_func (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/_extends/graph_kernel/model/
Dmodel.py103 def __init__(self, iter_type, calibrate=1, relation_func=None): argument
106 self.relation_func = relation_func
107 if relation_func is None:
108 self.relation_func = lambda *x: self.default_relation_func[iter_type](self, *x)
255 return cls.get_prim(op).relation_func(op, input_idx)
289 _, elem_relation = cls.get_prim(op).relation_func(op, input_idx)