Searched refs:HigherPrecisionElementType (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 438 element_type = ShapeUtil::HigherPrecisionElementType(*shape, *arg_shape); in InferConcatOpShape()
558 ShapeUtil::HigherPrecisionElementType(operand_shape, padding_value_shape), in InferPadShape()
748 ShapeUtil::HigherPrecisionElementType(lhs, rhs), dimensions); in InferDotOpShape()
779 return ShapeUtil::MakeShape(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferDegenerateDimensionBroadcastShape()
843 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()
916 lhs, ShapeUtil::HigherPrecisionElementType(lhs, rhs)); in InferElementwiseBinaryOpShape()
1742 return ShapeUtil::MakeShape(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferConvolveShape()
2408 on_true, ShapeUtil::HigherPrecisionElementType(on_true, on_false)); in InferSelectShape()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 231 static PrimitiveType HigherPrecisionElementType(const Shape& a, in HigherPrecisionElementType() function
|