Searched refs:HigherPrecisionElementType (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convert_operand_folding.cc | 27 ShapeUtil::HigherPrecisionElementType(hlo->operand(0)->shape(), in IsUpcastConvert()
|
| D | shape_inference.cc | 429 element_type = ShapeUtil::HigherPrecisionElementType(*shape, *arg_shape); in InferConcatOpShape()
580 ShapeUtil::HigherPrecisionElementType(operand_shape, padding_value_shape), in InferPadShape()
731 MaybeUpcast(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferDotOpShape()
774 return ShapeUtil::MakeShape(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferDegenerateDimensionBroadcastShape()
837 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()
923 lhs, ShapeUtil::HigherPrecisionElementType(lhs, rhs)); in InferElementwiseBinaryOpShape()
1868 MaybeUpcast(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferConvolveShape()
3153 pred, ShapeUtil::HigherPrecisionElementType(on_true, on_false)); in InferSelectShape()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 273 static PrimitiveType HigherPrecisionElementType(const Shape& a, in HigherPrecisionElementType() function
|
| D | shape_util.cc | 1739 HigherPrecisionElementType(from, to) == to.element_type(); in ElementCanUpcast()
|