Home | History | Annotate | Download | only in lib

Lines Matching refs:xla

19 #include "tensorflow/compiler/xla/client/computation.h"
20 #include "tensorflow/compiler/xla/client/computation_builder.h"
21 #include "tensorflow/compiler/xla/statusor.h"
27 xla::ComputationDataHandle Zeros(xla::ComputationBuilder* builder,
28                                  const xla::Shape& shape);
32 xla::ComputationDataHandle FloatLiteral(xla::ComputationBuilder* builder,
33                                         xla::PrimitiveType type, double value);
37 xla::ComputationDataHandle IntegerLiteral(xla::ComputationBuilder* builder,
38                                           xla::PrimitiveType type, int64 value);
41 xla::StatusOr<xla::ComputationDataHandle> SliceInMinorDims(
42     xla::ComputationBuilder* builder, const xla::ComputationDataHandle& x,
47 xla::StatusOr<xla::ComputationDataHandle> UpdateSlice(
48     xla::ComputationBuilder* builder, const xla::ComputationDataHandle& x,
49     const xla::ComputationDataHandle& update, gtl::ArraySlice<int64> start);
53 xla::StatusOr<xla::ComputationDataHandle> UpdateSliceInMinorDims(
54     xla::ComputationBuilder* builder, const xla::ComputationDataHandle& x,
55     const xla::ComputationDataHandle& update, gtl::ArraySlice<int64> start);
58 xla::StatusOr<xla::ComputationDataHandle> TransposeInMinorDims(
59     xla::ComputationBuilder* builder, const xla::ComputationDataHandle& x);