Searched refs:SameElementType (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 214 static bool SameElementType(const Shape& lhs, const Shape& rhs) { in SameElementType() function
225 return ShapeUtil::SameElementType(a, b); in SameElementTypeIgnoringFpPrecision()
233 if (SameElementType(a, b)) { in HigherPrecisionElementType()
|
| D | shape_util.cc | 110 if (!ShapeUtil::SameElementType(lhs, rhs)) { in CompareShapes()
620 return SameDimensions(lhs, rhs) && SameElementType(lhs, rhs); in Compatible()
|
| D | literal_util.cc | 446 TF_RET_CHECK(ShapeUtil::SameElementType(src_literal.shape(), shape())); in CopySliceFrom()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 899 CHECK(ShapeUtil::SameElementType(lhs_shape, rhs_shape)); in HandleConvolution()
900 CHECK(ShapeUtil::SameElementType(lhs_shape, result_shape)); in HandleConvolution()
1038 CHECK(ShapeUtil::SameElementType(lhs->shape(), rhs->shape())); in HandleDot()
1039 CHECK(ShapeUtil::SameElementType(lhs->shape(), dot->shape())); in HandleDot()
|
| D | algebraic_simplifier.cc | 1402 ShapeUtil::SameElementType(lhs->operand(1)->shape(), rhs->shape())) { in HandlePower()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2918 ShapeUtil::SameElementType(operands[0]->shape(), operand->shape())); in ElementTypesSameAndSupported()
|