diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-08 08:26:52 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-08 08:31:49 -0800 |
commit | a47cd30d960b128e5ed405cb36e914aa36fe462a (patch) | |
tree | fa019ded87328d5d3bf4a0ccc3b8721571006437 /tensorflow/compiler/jit/xla_compilation_cache.h | |
parent | 16a6666c1c1a3f4b288472c4f461b6418bda0170 (diff) |
This creates a new helper, xla_launch_util, that contains the business logic
of launching an XLA computation.
Also changes the resource variable container from a std::vector<OptionalTensor>
to a std::map<int, OptionalTensor> in preparation for backends where the
resource variables aren't ordered densely at the end of the argument list.
PiperOrigin-RevId: 188335574
Diffstat (limited to 'tensorflow/compiler/jit/xla_compilation_cache.h')
-rw-r--r-- | tensorflow/compiler/jit/xla_compilation_cache.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/jit/xla_compilation_cache.h b/tensorflow/compiler/jit/xla_compilation_cache.h index 0858020716..d506378314 100644 --- a/tensorflow/compiler/jit/xla_compilation_cache.h +++ b/tensorflow/compiler/jit/xla_compilation_cache.h @@ -63,7 +63,7 @@ class XlaCompilationCache : public ResourceBase { // outputs. Status Compile(const XlaCompiler::Options& options, const NameAttrList& function, int num_constant_args, - const std::vector<OptionalTensor>& variable_args, + const std::map<int, OptionalTensor>& variable_args, OpKernelContext* ctx, const XlaCompiler::CompilationResult** compilation_result, xla::LocalExecutable** executable, @@ -105,7 +105,7 @@ class XlaCompilationCache : public ResourceBase { // Builds the signature for a compilation. Status BuildSignature(const NameAttrList& function, int num_constant_args, - const std::vector<OptionalTensor>& variable_args, + const std::map<int, OptionalTensor>& variable_args, OpKernelContext* ctx, Signature* signature); // The value associated with a cache entry. |