Searched refs:HasCompatibleElementTypes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_traits.h | 55 if (!mlir::TF::HasCompatibleElementTypes(type, result_type)) in verifyTrait()
60 if (!mlir::TF::HasCompatibleElementTypes( in verifyTrait()
|
| D | tf_types.h | 296 bool HasCompatibleElementTypes(Type lhs, Type rhs,
|
| D | tf_types.cc | 347 bool HasCompatibleElementTypes(Type lhs, Type rhs, in HasCompatibleElementTypes() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.h | 192 bool HasCompatibleElementTypes(const Shape& shape_0, const Shape& shape_1,
|
| D | hlo_verifier.cc | 447 bool ShapeVerifier::HasCompatibleElementTypes(const Shape& shape_0, in HasCompatibleElementTypes() function in xla::ShapeVerifier
468 if (!HasCompatibleElementTypes(shape_0, shape_1, instruction->shape())) { in HandleRng()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1494 if (HasCompatibleElementTypes(input.getType(), result.getType())) { in InferShapeForFunctionReturnType()
|