Home
last modified time | relevance | path

Searched refs:ElementIsFloating (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvert_operand_folding.cc23 ShapeUtil::ElementIsFloating(hlo->shape()) == in IsUpcastConvert()
24 ShapeUtil::ElementIsFloating(hlo->operand(0)->shape()) && in IsUpcastConvert()
Dar_crs_combiner.cc147 return ShapeUtil::ElementIsFloating(instruction->shape()) == in MatchesArCrsPattern()
148 ShapeUtil::ElementIsFloating(instruction->operand(0)->shape()); in MatchesArCrsPattern()
153 return ShapeUtil::ElementIsFloating(instruction->shape()); in MatchesArCrsPattern()
Dshape_inference.cc269 if (!ShapeUtil::ElementIsFloating(shape)) { in InferUnaryOpShape()
287 if (!ShapeUtil::ElementIsFloating(shape) && in InferUnaryOpShape()
299 } else if (ShapeUtil::ElementIsFloating(shape)) { in InferUnaryOpShape()
329 !ShapeUtil::ElementIsFloating(shape) && in InferUnaryOpShape()
366 if (!ShapeUtil::ElementIsFloating(shape)) { in InferUnaryOpShape()
508 if (!ShapeUtil::ElementIsFloating(operand_shape)) { in InferReducePrecisionShape()
999 if (!ShapeUtil::ElementIsFloating(lhs)) { in InferBinaryOpShape()
1263 if (!ShapeUtil::ElementIsFloating(operand_shape)) { in InferBatchNormTrainingShape()
1369 if (!ShapeUtil::ElementIsFloating(operand_shape)) { in InferBatchNormInferenceShape()
1515 if (!ShapeUtil::ElementIsFloating(operand_shape)) { in InferBatchNormGradShape()
[all …]
Dhlo_verifier.cc1110 if (!ShapeUtil::ElementIsFloating(subshape)) { in CheckMixedPrecisionOperands()
Dalgebraic_simplifier.cc260 ShapeUtil::ElementIsFloating(src_shape) && in IsConvertPairNoOp()
261 ShapeUtil::ElementIsFloating(intermediate_shape); in IsConvertPairNoOp()
2439 ShapeUtil::ElementIsFloating(dot->shape()) in HandleDot()
Dhlo_evaluator.cc2363 bool use_fast_add = ShapeUtil::ElementIsFloating(init_values[0]->shape()) && in GenerateReduceOutputElement()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h264 if (ElementIsFloating(a) && ElementIsFloating(b)) { in SameElementTypeIgnoringFpPrecision()
281 ElementIsFloating(shape) in HigherPrecisionElementType()
286 ElementIsFloating(shape) in HigherPrecisionElementType()
503 static bool ElementIsFloating(const Shape& shape);
Dshape_util.cc481 /* static */ bool ShapeUtil::ElementIsFloating(const Shape& shape) { in ElementIsFloating() function in xla::ShapeUtil
1737 return ElementIsFloating(from) == ElementIsFloating(to) && in ElementCanUpcast()
Dliteral_comparison.cc813 if (ShapeUtil::ElementIsFloating(expected.shape()) || in NearHelper()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc317 if (ShapeUtil::ElementIsFloating(expected.shape()) || in ComputeAndCompareLiteralWithStatus()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1387 bool root_is_floating_point = ShapeUtil::ElementIsFloating(root_shape); in MatchReductionGenerator()