message: Allow all argument types except TensorList in XlaCompilerCache
  With recent changes, resource argument types are supported with MLIR compilation.
diff: @@ -278,16 +278,14 @@ Status XlaCompilationCache::CompileSingleOp(
  const NodeDef& node_def = ctx->op_kernel().def();
  TF_ASSIGN_OR_RETURN(auto graph, CreateGraph(node_def, args, result_dtypes));
  - bool are_args_supported =
  - absl::c_all_of(args, [](const XlaCompiler::Argument arg) {
  - return arg.kind == XlaCompiler::Argument::kConstant ||
  - arg.kind == XlaCompiler::Argument::kParameter;
  + bool has_tensor_list_arg =
  + absl::c_any_of(args, [](const XlaCompiler::Argument arg) {
  + return arg.kind == XlaCompiler::Argument::kTensorList;
  });
  const ConfigProto* config = ctx->function_library()->config_proto();
  bool use_mlir = config && config->experimental().enable_mlir_bridge();
  - // TODO(b/155596779): Understand the source of other argument types and
  - // depending on the source either support those or avoid these codepath.
  - if (!use_mlir || !are_args_supported) {
  + // TODO(b/155596779): Support TensorList args.
  + if (!use_mlir || !has_tensor_list_arg) {
  return compiler->CompileGraph(compile_options, node_def.name(),
  std::move(graph), args, result);
  }
  