message: [tf2xla] Add DT_INT64 to kEinsumTypes
  This allows einsum to be used with DT_INT64 operands.
diff: @@ -26,9 +26,9 @@ limitations under the License.
  namespace tensorflow {
  namespace {
  -constexpr std::array<DataType, 7> kEinsumTypes = {
  - {DT_INT32, DT_HALF, DT_BFLOAT16, DT_FLOAT, DT_DOUBLE, DT_COMPLEX64,
  - DT_COMPLEX128}};
  +constexpr std::array<DataType, 9> kEinsumTypes = {
  + {DT_INT32, DT_INT64, DT_UINT64, DT_HALF, DT_BFLOAT16, DT_FLOAT, DT_DOUBLE,
  + DT_COMPLEX64, DT_COMPLEX128}};
  // Kernel which compiles XlaEinsum, an einsum op accepting two inputs.
  class XlaEinsumOp : public XlaOpKernel {
  