Searched defs:bias_p (Results 1 – 2 of 2) sorted by relevance
245 const int32_t* const bias_p = in PrePackConvWeights() local
1294 const Tensor& bias_p, in convolution_jvp()1332 const Tensor& bias_p, in _convolution_jvp()6194 const Tensor& bias_p, in batch_norm_jvp()6249 const Tensor& bias_p, in layer_norm_jvp()6288 const Tensor& bias_p, in group_norm_jvp()