Using symbolic tensor with fixed tensor in Keras batchDot

Question

For my computational graph defined in Keras, I have two Tensors (BS X M X EMBED_DIM) and (BS X K X EMBED_DIM) and am trying to run a tf.keras