Searched defs:bias_v (Results 1 – 7 of 7) sorted by relevance
789 const float32x4_t bias_v = vdupq_n_f32(bias[0]); in av1_cnn_convolve_no_maxpool_padding_valid_2x2_large_neon() local888 const float32x4_t bias_v = vdupq_n_f32(*bias); in av1_cnn_convolve_no_maxpool_padding_valid_2x2_neon() local967 const float32x4_t bias_v = vdupq_n_f32(*bias); in av1_cnn_convolve_no_maxpool_padding_valid_5x5_neon() local
46 Tensor bias_v, in MultiheadAttentionForwardFuncOptions()
859 Tensor bias_v; variable
46 float bias_v = bias_data ? bias_data[c] : 0; in compute_fused_params() local
66 param_t bias_v = bias_data ? bias_data[c] : param_t(0); in batch_norm_cpu_collect_linear_and_constant_terms() local
3594 torch::Tensor bias_v; in _multihead_attn_test_helper() local
14916 bias_v, # type: Optional[Tensor] argument