Searched refs:getBroadcastedType (Results 1 – 12 of 12) sorted by relevance
| /external/llvm-project/mlir/include/mlir/Dialect/ |
| D | Traits.h | 74 Type getBroadcastedType(Type type1, Type type2, Type elementType = nullptr);
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | validators.cc | 105 return OpTrait::util::getBroadcastedType(a.getType(), b.getType()) != Type(); in IsBroadcastableElementsAttrs()
|
| /external/llvm-project/mlir/lib/Dialect/ |
| D | Traits.cpp | 106 Type OpTrait::util::getBroadcastedType(Type type1, Type type2, in getBroadcastedType() function in OpTrait::util
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 93 return OpTrait::util::getBroadcastedType(a, b) != Type(); in IsBroadcastableElementsAttrAndType()
103 OpTrait::util::getBroadcastedType(a, b, output_element_type); in OperandsBroadcastToOutputType()
|
| D | prepare_tf.cc | 1118 if (!OpTrait::util::getBroadcastedType(x_type, y_type)) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 502 Type broadcasted_type = OpTrait::util::getBroadcastedType( in Verify()
619 Type broadcasted_type = OpTrait::util::getBroadcastedType( in Verify()
|
| D | tf_op_base.td | 586 OpTrait::util::getBroadcastedType(x.getType(), y.getType());
|
| D | tf_ops_n_z.cc | 815 OpTrait::util::getBroadcastedType(e.getType(), t.getType()); in InferSelectV2OpType()
1170 auto broadcasted_ty = OpTrait::util::getBroadcastedType( in Verify()
|
| D | tf_ops_helpers.inc | 75 OpTrait::util::getBroadcastedType(x.getType(), y.getType());
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 389 auto type = OpTrait::util::getBroadcastedType(lhs.getType(), rhs.getType()) in ConstFoldBinaryOpDenseDense()
521 OpTrait::util::getBroadcastedType(lhs.getType(), rhs.getType()); in buildComparisonBinOp()
541 OpTrait::util::getBroadcastedType(lhs.getType(), rhs.getType()); in buildFusedBroadcastableBinOp()
2375 OpTrait::util::getBroadcastedType(x.getType(), y.getType()); in BuildSelectV2Op()
|
| D | tfl_ops.td | 432 OpTrait::util::getBroadcastedType(lhs.getType(), rhs.getType());
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | OpBase.td | 2260 CPred<"::mlir::OpTrait::util::getBroadcastedType("
|