Searched refs:bias_mat (Results 1 – 2 of 2) sorted by relevance
297 def construct(self, input_feature, bias_mat): argument305 logits = self.coef_activation(logits) + bias_mat384 def construct(self, input_data, bias_mat): argument387 res += (self.attns[i](input_data, bias_mat),)
115 def construct(self, input_data, bias_mat): argument117 input_data = cell(input_data, bias_mat)