Searched defs:XlaReductionOp (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dreduction_ops.h34 class XlaReductionOp : public XlaOpKernel { class in namespace:tensorflow
36 explicit XlaReductionOp(OpKernelConstruction* ctx);
37 ~XlaReductionOp() override {}
H A Dreduction_ops_common.cc27 XlaReductionOp::XlaReductionOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { function in class:tensorflow::XlaReductionOp
35 xla::ComputationDataHandle XlaReductionOp::InitialValue(
42 xla::ComputationDataHandle XlaReductionOp::BuildFinalizer(
49 void XlaReductionOp::Compile(XlaOpKernelContext* ctx) {

Completed in 127 milliseconds