Searched refs:EinsumDense (Results 1 – 8 of 8) sorted by relevance
245 layer = einsum_dense.EinsumDense(274 einsum_dense.EinsumDense,283 layer = einsum_dense.EinsumDense(291 layer = einsum_dense.EinsumDense(300 layer = einsum_dense.EinsumDense(equation="ab,bc->cd", output_shape=64)308 layer = einsum_dense.EinsumDense(equation="ab,zd->ad", output_shape=64)
34 class EinsumDense(Layer): class125 super(EinsumDense, self).__init__(**kwargs)167 super(EinsumDense, self).build(input_shape)197 base_config = super(EinsumDense, self).get_config()
350 self._query_dense = einsum_dense.EinsumDense(359 self._key_dense = einsum_dense.EinsumDense(368 self._value_dense = einsum_dense.EinsumDense(389 self._output_dense = einsum_dense.EinsumDense(
139 from tensorflow.python.keras.layers.einsum_dense import EinsumDense
4 name: "EinsumDense"
1 path: "tensorflow.keras.layers.experimental.EinsumDense"3 is_instance: "<class \'tensorflow.python.keras.layers.einsum_dense.EinsumDense\'>"