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::__anondd4c53d00111::TpuApproxTopKOp