Home | History | Annotate | Download | only in lib

Lines Matching refs:xla

21 #include "tensorflow/compiler/xla/client/computation.h"
22 #include "tensorflow/compiler/xla/client/computation_builder.h"
23 #include "tensorflow/compiler/xla/statusor.h"
27 // Builds an XLA computation that performs a scatter operation on `buffer`,
42 xla::StatusOr<xla::ComputationDataHandle> XlaScatter(
43     const xla::ComputationDataHandle& buffer,
44     const xla::ComputationDataHandle& updates,
45     const xla::ComputationDataHandle& indices, bool indices_are_vectors,
46     const std::function<xla::ComputationDataHandle(
47         xla::ComputationDataHandle, xla::ComputationDataHandle,
48         xla::ComputationBuilder*)>& combiner,
49     xla::ComputationBuilder* builder);