Searched refs:createOrFold (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | broadcast_utils.cc | 62 Value lhs_shape_v = builder.createOrFold<shape::ShapeOfOp>(loc, lhs); in ComputeBinaryElementwiseBroadcastingResultExtents()
63 Value rhs_shape_v = builder.createOrFold<shape::ShapeOfOp>(loc, rhs); in ComputeBinaryElementwiseBroadcastingResultExtents()
67 Value result_shape_v = builder.createOrFold<shape::BroadcastOp>( in ComputeBinaryElementwiseBroadcastingResultExtents()
70 return builder.createOrFold<tensor::CastOp>( in ComputeBinaryElementwiseBroadcastingResultExtents()
75 return builder.createOrFold<shape::BroadcastOp>( in ComputeBinaryElementwiseBroadcastingResultExtents()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | materialize_broadcasts.cc | 52 max_value = rewriter.createOrFold<BroadcastOp>( in matchAndRewrite()
60 min_value = rewriter.createOrFold<BroadcastOp>( in matchAndRewrite()
|
| D | unfuse_batch_norm.cc | 44 return rewriter.createOrFold<mhlo::DynamicBroadcastInDimOp>( in BroadcastToFeatureDim()
101 return rewriter.createOrFold<mhlo::DynamicBroadcastInDimOp>( in MaterializeEpsilon()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 320 builder.createOrFold<mlir::AffineLoadOp>( in CreateNaiveMlirConv()
339 builder.createOrFold<mlir::AffineLoadOp>( in CreateNaiveMlirConv()
345 builder.createOrFold<mlir::AffineLoadOp>(location, output_acc); in CreateNaiveMlirConv()
346 builder.createOrFold<mlir::AffineStoreOp>( in CreateNaiveMlirConv()
360 builder.createOrFold<mlir::AffineStoreOp>( in CreateNaiveMlirConv()
364 builder.createOrFold<mlir::AffineLoadOp>(location, output_acc), in CreateNaiveMlirConv()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 409 void createOrFold(SmallVectorImpl<Value> &results, Location location, in createOrFold() function
432 createOrFold(Location location, Args &&... args) { in createOrFold() function
434 createOrFold<OpTy>(results, location, std::forward<Args>(args)...); in createOrFold()
442 createOrFold(Location location, Args &&... args) { in createOrFold() function
|
| /external/llvm-project/mlir/docs/ |
| D | Canonicalization.md | 98 `OpBuilder::createOrFold`.
|
| /external/llvm-project/mlir/lib/Conversion/VectorToSCF/ |
| D | VectorToSCF.cpp | 56 res.push_back(b.createOrFold<AffineApplyOp>(loc, map, operands)); in applyMapToValues()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 126 return rewriter.createOrFold<TF::CastOp>(loc, new_type, val, in CreateCastToInt32()
|
| /external/llvm-project/mlir/test/lib/Dialect/Test/ |
| D | TestDialect.cpp | 711 builder.createOrFold<DimOp>(getLoc(), getOperand(0), 0)}; in reifyReturnTypeShapes()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 41 return b.createOrFold<AffineApplyOp>(loc, map, operands); in createFoldedComposedAffineApply()
|