Home
last modified time | relevance | path

Searched refs:SameElementTypeIgnoringFpPrecision (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc73 ShapeUtil::SameElementTypeIgnoringFpPrecision(input_shape, kernel_shape)); in PotentiallyImplementedAsEigenConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc407 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(*arg_shape, *shape)) { in InferConcatOpShape()
547 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(operand_shape, in InferPadShape()
901 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(lhs, rhs)) { in InferElementwiseBinaryOpShape()
1131 if (ShapeUtil::SameElementTypeIgnoringFpPrecision(*arg_shapes[i], in InferMapShape()
1196 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(parameter_shape, in InferMapShape()
1270 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(offset_shape, in InferBatchNormTrainingShape()
1280 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(scale_shape, in InferBatchNormTrainingShape()
1376 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(offset_shape, in InferBatchNormInferenceShape()
1387 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(scale_shape, in InferBatchNormInferenceShape()
1398 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(mean_shape, in InferBatchNormInferenceShape()
[all …]
Dhlo_verifier.h173 ? ShapeUtil::SameElementTypeIgnoringFpPrecision(a, b) in SameElementType()
Dhlo_verifier.cc453 ShapeUtil::SameElementTypeIgnoringFpPrecision(shape_0, in HasCompatibleElementTypes()
/external/tensorflow/tensorflow/compiler/xla/
Dshape.cc137 !ShapeUtil::SameElementTypeIgnoringFpPrecision(lhs, rhs)) || in operator ()()
Dshape_util.h262 static bool SameElementTypeIgnoringFpPrecision(const Shape& a, in SameElementTypeIgnoringFpPrecision() function