Searched refs:SameElementTypeIgnoringFpPrecision (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 373 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(*arg_shape, *shape)) { in InferConcatOpShape()
502 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(operand_shape, in InferPadShape()
625 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(lhs, rhs)) { in InferDotOpShape()
866 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(lhs, rhs)) { in InferElementwiseBinaryOpShape()
1087 if (ShapeUtil::SameElementTypeIgnoringFpPrecision(*arg_shapes[i], in InferMapShape()
1152 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(parameter_shape, in InferMapShape()
1226 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(offset_shape, in InferBatchNormTrainingShape()
1236 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(scale_shape, in InferBatchNormTrainingShape()
1332 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(offset_shape, in InferBatchNormInferenceShape()
1343 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(scale_shape, in InferBatchNormInferenceShape()
[all …]
|
| D | hlo_verifier.h | 139 ? ShapeUtil::SameElementTypeIgnoringFpPrecision(a, b) in SameElementType()
|
| D | hlo_verifier.cc | 278 ShapeUtil::SameElementTypeIgnoringFpPrecision(shape_0, in HasCompatibleElementTypes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 73 ShapeUtil::SameElementTypeIgnoringFpPrecision(input_shape, kernel_shape)); in PotentiallyImplementedAsEigenConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape.cc | 137 !ShapeUtil::SameElementTypeIgnoringFpPrecision(lhs, rhs)) || in operator ()()
|
| D | shape_util.h | 256 static bool SameElementTypeIgnoringFpPrecision(const Shape& a, in SameElementTypeIgnoringFpPrecision() function
272 CHECK(SameElementTypeIgnoringFpPrecision(a, b)); in HigherPrecisionElementType()
|