Searched defs:PackJITCompileOpPattern (Results 1 – 1 of 1) sorted by relevance
183 struct PackJITCompileOpPattern struct187 explicit PackJITCompileOpPattern(MLIRContext *ctx, in PackJITCompileOpPattern() function201 LogicalResult matchAndRewrite(tf_framework::JITCompileOp op, in matchAndRewrite()243 llvm::ArrayRef<int64_t> tile_sizes;244 llvm::ArrayRef<int64_t> unroll_factors;245 int64_t max_supported_rank;246 bool enable_ftz;247 bool index_64bit_if_jit_compiling;248 bool cpu_codegen;