Searched defs:bias_dim (Results 1 – 4 of 4) sorted by relevance
208 const auto bias_dim = attn_bias_.value().dim(); in _scaled_dot_product_cudnn_attention_backward_cuda() local
229 auto bias_dim = static_cast<const ConcreteType *>(this)->getBiasDim( in matchAndRewrite() local
827 auto bias_dim = filter_type.getShape().back(); in matchAndRewrite() local916 auto bias_dim = filter_shape[2] * filter_shape[3]; in matchAndRewrite() local
397 DimensionHandle bias_dim = c->Dim(bias_shape, 0); in BiasAddShape() local