Lines Matching refs:xla
16 // XLA-specific base classes for Reduction Ops.
22 #include "tensorflow/compiler/xla/client/computation_builder.h"
31 // xla::ComputationBuilder and BuildFinalizer adds the
33 // xla::ComputationBuilder.
40 virtual xla::ComputationDataHandle InitialValue(
41 xla::ComputationBuilder* builder);
47 virtual void BuildReducer(xla::ComputationBuilder* builder,
48 const xla::ComputationDataHandle& scalar_lhs,
49 const xla::ComputationDataHandle& scalar_rhs) = 0;
56 virtual xla::ComputationDataHandle BuildFinalizer(
57 xla::ComputationBuilder* builder,
58 const xla::ComputationDataHandle& reduce_output,
