Searched refs:add_weight (Results 1 – 6 of 6) sorted by relevance
26 def __init__(self, matmul_weight, add_weight, strategy1=None, strategy2=None): argument31 self.bias = Parameter(add_weight, "bias")
131 … self.add_weight = Parameter(Tensor(np.ones([128, 32]), dtype=ms.float32), name="weight1")139 out = out + self.add_weight
53 void add_weight(uint16_t weight) { in add_weight() function76 builder_.add_weight(weight);
138 auto add_weight = add_cnode->input(kInputIndexTwo); in CheckCanFusion() local139 MS_ASSERT(add_weight != nullptr); in CheckCanFusion()141 if (FetchShapeFromAbstract(add_weight->abstract(), &shape) != lite::RET_OK) { in CheckCanFusion()
203 virtual ParameterPtr add_weight(const tensor::MetaTensorPtr &meta_tensor);
725 ParameterPtr FuncGraph::add_weight(const tensor::MetaTensorPtr &meta_tensor) { in add_weight() function in mindspore::FuncGraph