Home
last modified time | relevance | path

Searched refs:VectorizationPattern (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_vectorize_tiled_ops.cc95 struct VectorizationPattern : public mlir::OpRewritePattern<OpTy> { struct
96 VectorizationPattern(MLIRContext *context, in VectorizationPattern() argument
151 patterns.add<VectorizationPattern<FillOp>>(ctx, isFillTiledOrSmall); in runOnOperation()
152 patterns.add<VectorizationPattern<GenericOp>>( in runOnOperation()