Searched refs:IsLegalNumpyRankedBroadcast (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo_patterns.td | 25 def IsLegalNumpyRankedBroadcast :
26 Constraint<CPred<"hlo::IsLegalNumpyRankedBroadcast($0, $1, $2)">,
65 [(IsLegalNumpyRankedBroadcast $l, $r, $broadcast_dimensions)]>;
74 [(IsLegalNumpyRankedBroadcast $l, $r, $broadcast_dimensions)]>;
82 [(IsLegalNumpyRankedBroadcast $l, $r, $broadcast_dimensions)]>;
89 [(IsLegalNumpyRankedBroadcast $l, $r, $broadcast_dimensions)]>;
94 [(IsLegalNumpyRankedBroadcast $l, $r, $broadcast_dimensions)]>;
100 [(IsLegalNumpyRankedBroadcast $l, $r, $broadcast_dimensions)]>;
190 [(IsLegalNumpyRankedBroadcast $l, $r, $broadcast_dimensions)]>;
202 [(IsLegalNumpyRankedBroadcast $l, $r, $broadcast_dimensions)]>;
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/utils/ |
| D | broadcast_utils.h | 36 bool IsLegalNumpyRankedBroadcast(Value lhs, Value rhs,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | broadcast_utils.cc | 31 bool IsLegalNumpyRankedBroadcast(Value lhs, Value rhs, in IsLegalNumpyRankedBroadcast() function
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_ops.cc | 169 !hlo::IsLegalNumpyRankedBroadcast(lhs, rhs, broadcast_dimensions)) { in ReifyBroadcastBinaryOpReturnTypeShapes()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 1067 !hlo::IsLegalNumpyRankedBroadcast(lhs, rhs, *broadcast_dimensions)) { in matchAndRewrite()
|