Lines Matching refs:xla
16 // XLA-specific reduction Ops.
22 #include "tensorflow/compiler/xla/literal_util.h"
31 void BuildReducer(xla::ComputationBuilder* builder,
32 const xla::ComputationDataHandle& scalar_lhs,
33 const xla::ComputationDataHandle& scalar_rhs) override {
44 xla::ComputationDataHandle InitialValue(
45 xla::ComputationBuilder* builder) override {
49 void BuildReducer(xla::ComputationBuilder* builder,
50 const xla::ComputationDataHandle& scalar_lhs,
51 const xla::ComputationDataHandle& scalar_rhs) override {
63 xla::ComputationDataHandle InitialValue(
64 xla::ComputationBuilder* builder) override {
65 xla::PrimitiveType type;
67 return builder->ConstantLiteral(xla::Literal::MaxValue(type));
70 void BuildReducer(xla::ComputationBuilder* builder,
71 const xla::ComputationDataHandle& scalar_lhs,
72 const xla::ComputationDataHandle& scalar_rhs) override {
83 xla::ComputationDataHandle InitialValue(
84 xla::ComputationBuilder* builder) override {
85 xla::PrimitiveType type;
87 return builder->ConstantLiteral(xla::Literal::MinValue(type));
90 void BuildReducer(xla::ComputationBuilder* builder,
91 const xla::ComputationDataHandle& scalar_lhs,
92 const xla::ComputationDataHandle& scalar_rhs) override {
103 void BuildReducer(xla::ComputationBuilder* builder,
104 const xla::ComputationDataHandle& scalar_lhs,
105 const xla::ComputationDataHandle& scalar_rhs) override {
109 xla::ComputationDataHandle BuildFinalizer(
110 xla::ComputationBuilder* builder,
111 const xla::ComputationDataHandle& reduce_output,
126 xla::ComputationDataHandle InitialValue(
127 xla::ComputationBuilder* builder) override {
131 void BuildReducer(xla::ComputationBuilder* builder,
132 const xla::ComputationDataHandle& scalar_lhs,
133 const xla::ComputationDataHandle& scalar_rhs) override {
144 xla::ComputationDataHandle InitialValue(
145 xla::ComputationBuilder* builder) override {
149 void BuildReducer(xla::ComputationBuilder* builder,
150 const xla::ComputationDataHandle& scalar_lhs,
151 const xla::ComputationDataHandle& scalar_rhs) override {
