Searched refs:TopKOp (Results 1 – 1 of 1) sorted by relevance
27 class TopKOp : public XlaOpKernel { class29 explicit TopKOp(OpKernelConstruction* context) : XlaOpKernel(context) { in TopKOp() function in tensorflow::__anon561944120111::TopKOp62 TopKOp);