Searched defs:weight_p (Results 1 – 1 of 1) sorted by relevance
1292 const Tensor& weight_p, in convolution_jvp()1330 const Tensor& weight_p, in _convolution_jvp()6157 const Tensor& weight_p, in _affine_jvp()6192 const Tensor& weight_p, in batch_norm_jvp()6247 const Tensor& weight_p, in layer_norm_jvp()6286 const Tensor& weight_p, in group_norm_jvp()