Searched refs:nested_loc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 210 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
395 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
524 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
567 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
578 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange args) { in matchAndRewrite() argument
866 [&](OpBuilder& nested_builder, Location nested_loc, ValueRange ivs, in matchAndRewrite()
869 nested_loc, ivs[iota_op.iota_dimension()], in matchAndRewrite()
873 cast_op = nested_builder.create<SIToFPOp>(nested_loc, cast_op, in matchAndRewrite()
876 nested_builder.create<linalg::YieldOp>(nested_loc, cast_op); in matchAndRewrite()
|