Searched refs:beta_broadcast_dims (Results 1 – 1 of 1) sorted by relevance
6046 auto beta_broadcast_dims = llvm::to_vector<4>(batch_dim_indices); in ComputeWYRepresentation() local6047 beta_broadcast_dims.push_back(n_index); in ComputeWYRepresentation()6050 GetI64ElementsAttr(beta_broadcast_dims, builder), *rewriter); in ComputeWYRepresentation()6066 auto beta_broadcast_dims = llvm::to_vector<4>(batch_dim_indices); in ComputeWYRepresentation() local6067 beta_broadcast_dims.push_back(n_index); in ComputeWYRepresentation()6069 loc, neg_beta, v, GetI64ElementsAttr(beta_broadcast_dims, rewriter), in ComputeWYRepresentation()