aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/python/local_computation_builder.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/python/local_computation_builder.h')
-rw-r--r--tensorflow/compiler/xla/python/local_computation_builder.h7
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;