Home
last modified time | relevance | path

Searched defs:LowerRankSpecializationClusterPattern (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Drank_specialization.cc855 struct LowerRankSpecializationClusterPattern struct
857 LowerRankSpecializationClusterPattern(MLIRContext *ctx, in LowerRankSpecializationClusterPattern() argument
862 LogicalResult matchAndRewrite(chlo::RankSpecializationClusterOp op, in matchAndRewrite()
901 int64_t max_target_rank;