[XLA] Teach Executable to do its own profiling (patch 1/4).

Presently, ExecuteOnStreamWrapper is a method on xla::Service, where it doesn't really conceptually belong -- note that it doesn't use anything from the containing Service object, but it does have an Executable object as its first parameter that it could easily be a method on instead.  The only reason that it needs to be on Service is that it needs to access a Backend object in order to call backend->compiler()->shape_size_function(), and simply moving that into Executable would introduce a dependency cycle.

Thus, this patch (the first part of a sequence to address this) teaches Executable and its derivatives to compute shape_size_function.  In the CPU cases, this is simply a static function.  However, in the GPU case, we need to pass in the shape_size_function to the constructor, since it depends on a pointer size computed in the GpuCompiler.

PiperOrigin-RevId: 156807318

0 files changed, 0 insertions, 0 deletions