Searched refs:HasCompatibleElementTypes (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_traits.h | 57 if (!mlir::tf_type::HasCompatibleElementTypes(type, result_type)) in verifyTrait()
62 if (!mlir::tf_type::HasCompatibleElementTypes( in verifyTrait()
|
| H A D | tf_types.h | 35 using ::mlir::tf_type::HasCompatibleElementTypes; // NOLINT
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/ |
| H A D | ops.td | 570 return tf_type::HasCompatibleElementTypes(lhs, rhs);
684 return tf_type::HasCompatibleElementTypes(lhs, rhs);
837 return tf_type::HasCompatibleElementTypes(lhs, rhs);
958 return tf_type::HasCompatibleElementTypes(lhs, rhs);
|
| H A D | ops.cc | 931 if (!tf_type::HasCompatibleElementTypes(arg_type, op_type)) { in VerifySignature()
941 if (!tf_type::HasCompatibleElementTypes(ret_type, res_type)) { in VerifySignature()
966 if (!tf_type::HasCompatibleElementTypes(type, in VerifyTypeArray()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/types/ |
| H A D | dialect.h | 256 bool HasCompatibleElementTypes(Type lhs, Type rhs,
|
| H A D | dialect.cc | 803 bool HasCompatibleElementTypes(Type lhs, Type rhs, in HasCompatibleElementTypes() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_verifier.h | 297 bool HasCompatibleElementTypes(const Shape& shape_0, const Shape& shape_1,
|
| H A D | hlo_verifier.cc | 797 bool ShapeVerifier::HasCompatibleElementTypes(const Shape& shape_0, in HasCompatibleElementTypes() function in xla::ShapeVerifier
818 if (!HasCompatibleElementTypes(shape_0, shape_1, instruction->shape())) { in HandleRng()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/region_to_functional/ |
| H A D | impl.cc | 724 if (!tf_type::HasCompatibleElementTypes(lhs.getType(), rhs.getType())) in RegionEqualTo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | shape_inference.cc | 2585 if (HasCompatibleElementTypes(input.getType(), result.getType()) && in InferShapeForFunctionReturnType()
|