Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
H A Dconv_emitter_transforms.cc95 llvm::SmallVector<mlir::AffineForOp, 4> all_loops; in TileLoop() local
H A Dconv_emitter.cc493 llvm::SmallVector<mlir::AffineForOp, 4> all_loops; in TransformMlirConv() local
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_reductions.cpp1033 auto all_loops = loop.getAllLoopNestsWritingToBuf(c.buf()); in TEST() local
1075 auto all_loops = loop.getAllLoopNestsWritingToBuf(c.buf()); in TEST() local
1687 auto all_loops = loop.getAllLoopNestsWritingToBuf(rfac_buf); in TEST() local
1752 auto all_loops = loop.getAllLoopNestsWritingToBuf(rfac_buf); in TEST() local
H A Dtest_llvm.cpp1484 auto all_loops = loopnest.getAllLoopNestsWritingToBuf(b.buf()); in TEST() local