Arg<TF_Float32Tensor, [{The embedding activations Tensor to return.}]>:$sliced_activations,

    ConfinedAttr<I64Attr, [IntMinValue<0>]>:$table_id,
    ConfinedAttr<I64Attr, [IntMinValue<0>]>:$lookup_id
  );

  let results = (outs
    TF_Float32Tensor:$output
  );
}

def TF_TPUExecuteOp : TF_Op<"TPUExecute", [DeclareOpInterfaceMethods<MemoryEffectsOpInterface>]> {