[TF:XLA] Change XlaCompilationCache to accept XlaCompiler::Arguments in most...

[TF:XLA] Change XlaCompilationCache to accept XlaCompiler::Arguments in most cases rather than parsing an OpKernelContext.

Build the compilation cache signature from the XlaCompiler::Arguments, rather than building it inside the cache. Add a helper method XlaComputationLaunchContext::BuildXlaCompilerArguments that builds the necessary XlaCompiler::Arguments.

Fix a bug in XlaCompilationCache::Signature::operator==, where only the contents of a tensor were compared for equality, not the shapes.

Remove code that reasons about OpKernelContexts in XlaCompiler::CompileSingleOp, instead passing the specific information required.

PiperOrigin-RevId: 220373606