Searched refs:HigherPrecisionElementType (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 395 element_type = ShapeUtil::HigherPrecisionElementType(*shape, *arg_shape); in InferConcatOpShape()
540 ShapeUtil::HigherPrecisionElementType(operand_shape, padding_value_shape), in InferPadShape()
703 ShapeUtil::HigherPrecisionElementType(lhs, rhs), dimensions, is_dynamic); in InferDotOpShape()
739 return ShapeUtil::MakeShape(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferDegenerateDimensionBroadcastShape()
802 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()
888 lhs, ShapeUtil::HigherPrecisionElementType(lhs, rhs)); in InferElementwiseBinaryOpShape()
1814 return ShapeUtil::MakeShape(ShapeUtil::HigherPrecisionElementType(lhs, rhs), in InferConvolveShape()
2793 on_true, ShapeUtil::HigherPrecisionElementType(on_true, on_false)); in InferSelectShape()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 267 static PrimitiveType HigherPrecisionElementType(const Shape& a, in HigherPrecisionElementType() function
|