Searched defs:PackJITCompileOpPattern (Results 1 – 1 of 1) sorted by relevance
147 struct PackJITCompileOpPattern struct151 explicit PackJITCompileOpPattern(MLIRContext *ctx, in PackJITCompileOpPattern() function165 LogicalResult matchAndRewrite(tf_framework::JITCompileOp op, in matchAndRewrite()208 llvm::ArrayRef<StringRef> architectures;209 llvm::ArrayRef<int64_t> tile_sizes;210 llvm::ArrayRef<int64_t> unroll_factors;211 int64_t max_supported_rank;212 bool enable_ftz;213 bool cpu_codegen;