Searched defs:set_bias (Results 1 – 6 of 6) sorted by relevance
65 virtual void set_bias(const std::optional<at::Tensor>& bias) { in set_bias() function
118 virtual void set_bias(std::optional<at::Tensor> /*bias*/) { in set_bias() function
20 void set_bias(torch::Tensor bias) { in set_bias() function
105 void set_bias(const AccumScalar* ptr) { storage_.bias = ptr; } in set_bias() function
487 void PackedLinearWeightFp16::set_bias(std::optional<at::Tensor> bias) { in set_bias() function in PackedLinearWeightFp16
1026 NormalizeDescriptor& set_bias(float bias) { in set_bias() function