Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc233 std::vector<mlir::AffineForOp> cartesian_product_loops; member
258 std::vector<mlir::AffineForOp> cartesian_product_loops = in CreateNaiveMlirConv() local
262 OpBuilder::atBlockTerminator(cartesian_product_loops.back().getBody()); in CreateNaiveMlirConv()
277 mlir::AffineForOp loop_n = cartesian_product_loops[0]; in CreateNaiveMlirConv()
278 mlir::AffineForOp loop_o = cartesian_product_loops[1]; in CreateNaiveMlirConv()
282 for (auto loop : absl::MakeSpan(cartesian_product_loops).subspan(2)) { in CreateNaiveMlirConv()
369 return InitialMlirConvAnchors{cartesian_product_loops, reduction_loops, in CreateNaiveMlirConv()
391 std::vector<mlir::AffineForOp> cartesian_product_loops; member
397 std::vector<mlir::AffineForOp> cartesian_product_loops = in TransformMlirConv() local
398 anchors.cartesian_product_loops; in TransformMlirConv()
[all …]