Lines Matching refs:xla
16 // XLA-specific base classes for Unary and Binary Ops.
22 #include "tensorflow/compiler/xla/client/client_library.h"
23 #include "tensorflow/compiler/xla/client/computation_builder.h"
33 // implementation of the operation to a xla::ComputationBuilder. For most
34 // arithmetic Ops XLA handles the broadcasting automatically given the input
52 // may be useful for Ops that can't use standard XLA automatic
57 // in the XLA documentation.
58 virtual xla::ComputationDataHandle Computation(
59 XlaOpKernelContext* ctx, const xla::ComputationDataHandle& lhs,
61 const xla::ComputationDataHandle& rhs,
70 static std::pair<xla::ComputationDataHandle, xla::ComputationDataHandle>
71 Broadcast(xla::ComputationBuilder* builder,
72 const xla::ComputationDataHandle& lhs,
73 const xla::ComputationDataHandle& rhs,
