Searched refs:HigherPrecisionElementType (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 410 element_type = ShapeUtil::HigherPrecisionElementType(*shape, *arg_shape); in InferConcatOpShape()
577 ShapeUtil::HigherPrecisionElementType(operand_shape, padding_value_shape), in InferPadShape()
735 MaybeUpcast(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferDotOpShape()
778 return ShapeUtil::MakeShape(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferDegenerateDimensionBroadcastShape()
841 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()
927 lhs, ShapeUtil::HigherPrecisionElementType(lhs, rhs)); in InferElementwiseBinaryOpShape()
1869 MaybeUpcast(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferConvolveShape()
3294 pred, ShapeUtil::HigherPrecisionElementType(on_true, on_false)); in InferSelectShape()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 183 static PrimitiveType HigherPrecisionElementType(const Shape& a, in HigherPrecisionElementType() function
|
| D | shape_util.cc | 1740 return HigherPrecisionElementType(from, to) == to.element_type(); in ElementCanUpcast()
|