Home
last modified time | relevance | path

Searched refs:HasCompatibleElementTypes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_traits.h57 if (!mlir::tf_type::HasCompatibleElementTypes(type, result_type)) in verifyTrait()
62 if (!mlir::tf_type::HasCompatibleElementTypes( in verifyTrait()
Dtf_types.h35 using ::mlir::tf_type::HasCompatibleElementTypes; // NOLINT
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.h254 bool HasCompatibleElementTypes(Type lhs, Type rhs,
Ddialect.cc740 bool HasCompatibleElementTypes(Type lhs, Type rhs, in HasCompatibleElementTypes() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h197 bool HasCompatibleElementTypes(const Shape& shape_0, const Shape& shape_1,
Dhlo_verifier.cc781 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/
Dshape_inference.cc1723 if (HasCompatibleElementTypes(input.getType(), result.getType()) && in InferShapeForFunctionReturnType()