Searched defs:TpuApproxTopKOp (Results 1 – 1 of 1) sorted by relevance
115 class TpuApproxTopKOp : public ApproxTopKOpBase { class117 explicit TpuApproxTopKOp(OpKernelConstruction* ctx) : ApproxTopKOpBase(ctx) {} in TpuApproxTopKOp() function in tensorflow::__anon7e60a0010111::TpuApproxTopKOp