Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/
Dlegalize_to_linalg_utils.h174 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc236 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
472 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
618 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
1048 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
1497 Operation* linalgOp = rewriter.create<linalg::BatchMatmulOp>( in matchAndRewrite() local
1527 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
1661 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
2585 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
2905 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
3025 auto linalgOp = rewriter.create<linalg::GenericOp>( in matchAndRewrite() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
Dfusion_interface_impl.cc258 linalg::LinalgOp linalgOp = genericOp; in fuse() local
Dbufferize_tiled_loop.cc206 linalg::LinalgOp linalgOp, in createLinalgOpOnBuffers()
Dtransforms.cc122 LinalgOp linalgOp, ArrayRef<Attribute> iteratorTypes, in generateLoopNest()