Searched defs:nestedLoc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt_passes.cc | 403 auto dot = [&](mlir::OpBuilder& builder, mlir::Location nestedLoc) { in matchAndRewrite()
415 auto vecmat = [&](mlir::OpBuilder& builder, mlir::Location nestedLoc) { in matchAndRewrite()
426 auto matvec = [&](mlir::OpBuilder& builder, mlir::Location nestedLoc) { in matchAndRewrite()
438 auto generic = [&](mlir::OpBuilder& builder, mlir::Location nestedLoc) { in matchAndRewrite()
451 [&](mlir::OpBuilder& builder, mlir::Location nestedLoc) { in matchAndRewrite()
455 [&](mlir::OpBuilder& builder, mlir::Location nestedLoc) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| D | transforms.cc | 136 ValueRange outputs) { in generateLoopNest()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 475 [reductionAxe](OpBuilder& b, Location nestedLoc, ValueRange args) { in matchAndRewrite()
1059 [&](OpBuilder& nestedBuilder, Location nestedLoc, ValueRange /*args*/) { in matchAndRewrite()
2240 [&](OpBuilder& nestedBuilder, Location nestedLoc, ValueRange) { in matchAndRewrite()
|