Searched refs:HasCompatibleElementTypes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_traits.h | 57 if (!mlir::tf_type::HasCompatibleElementTypes(type, result_type)) in verifyTrait()
62 if (!mlir::tf_type::HasCompatibleElementTypes( in verifyTrait()
|
| D | tf_types.h | 35 using ::mlir::tf_type::HasCompatibleElementTypes; // NOLINT
|
| /external/tensorflow/tensorflow/core/ir/types/ |
| D | dialect.h | 254 bool HasCompatibleElementTypes(Type lhs, Type rhs,
|
| D | dialect.cc | 740 bool HasCompatibleElementTypes(Type lhs, Type rhs, in HasCompatibleElementTypes() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.h | 197 bool HasCompatibleElementTypes(const Shape& shape_0, const Shape& shape_1,
|
| D | hlo_verifier.cc | 781 bool ShapeVerifier::HasCompatibleElementTypes(const Shape& shape_0, in HasCompatibleElementTypes() function in xla::ShapeVerifier
802 if (!HasCompatibleElementTypes(shape_0, shape_1, instruction->shape())) { in HandleRng()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1723 if (HasCompatibleElementTypes(input.getType(), result.getType()) && in InferShapeForFunctionReturnType()
|