diff options
Diffstat (limited to 'tensorflow/compiler/xla/python/local_computation_builder.h')
-rw-r--r-- | tensorflow/compiler/xla/python/local_computation_builder.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/python/local_computation_builder.h b/tensorflow/compiler/xla/python/local_computation_builder.h index a7375c8965..31046e60f1 100644 --- a/tensorflow/compiler/xla/python/local_computation_builder.h +++ b/tensorflow/compiler/xla/python/local_computation_builder.h @@ -268,6 +268,13 @@ class LocalComputationBuilder { const ComputationDataHandle& false_operand, const LocalComputation& false_computation); + StatusOr<bool> IsConstant(const ComputationDataHandle& operand, + int64 num_parameters); + + StatusOr<std::unique_ptr<Literal> > ComputeConstant( + const ComputationDataHandle& operand, const Layout* output_layout, + tensorflow::gtl::ArraySlice<Literal> parameters); + #define _FORWARD(method_name, return_sig, args_sig) \ return_sig method_name args_sig; |