Searched refs:is_broadcastable (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/test/Dialect/Shape/ |
| D | ops.mlir | 266 %result = shape.is_broadcastable %a, %b
273 %result = shape.is_broadcastable %a, %b
|
| /external/llvm-project/mlir/include/mlir/Dialect/Shape/IR/ |
| D | ShapeOps.td | 186 def Shape_IsBroadcastableOp : Shape_Op<"is_broadcastable", [Commutative]> {
193 Concretely, shape.is_broadcastable returning true implies that
200 %true = shape.is_broadcastable [2,2], [3,1,2]
201 %false = shape.is_broadcastable [2,2], [3,2]
|
| /external/llvm-project/mlir/test/Conversion/ShapeToStandard/ |
| D | shape-to-standard.mlir | 389 %0 = shape.is_broadcastable %a, %b : tensor<3xindex>, tensor<?xindex>
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 406 auto is_broadcastable = [this, &properties](const string& input) { in HasAllInputsBroadcastableToShape() local
413 is_broadcastable); in HasAllInputsBroadcastableToShape()
|