Searched defs:ApproxTopKOpBase (Results 1 – 1 of 1) sorted by relevance
52 class ApproxTopKOpBase : public XlaOpKernel { class54 explicit ApproxTopKOpBase(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ApproxTopKOpBase() function in tensorflow::__anon7e60a0010111::ApproxTopKOpBase