Searched defs:OptimizeTopK (Results 1 – 1 of 1) sorted by relevance
1527 struct OptimizeTopK : public OpRewritePattern<TFL::TopKV2Op> { struct1532 llvm::Optional<int32_t> ComputeSliceK(Value value) const { in ComputeSliceK()1570 LogicalResult matchAndRewrite(TFL::TopKV2Op op, in matchAndRewrite()