Searched refs:XlaMergeOp (Results 1 – 2 of 2) sorted by relevance
130 class XlaMergeOp : public OpKernel {132 explicit XlaMergeOp(OpKernelConstruction* ctx);
547 XlaMergeOp::XlaMergeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in XlaMergeOp() function in tensorflow::XlaMergeOp549 void XlaMergeOp::Compute(OpKernelContext* ctx) { in Compute()578 REGISTER_KERNEL_BUILDER(Name("_XlaMerge").Device(DEVICE_CPU), XlaMergeOp);579 REGISTER_KERNEL_BUILDER(Name("_XlaMerge").Device(DEVICE_GPU), XlaMergeOp);