Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dar_crs_combiner.cc55 return ShapeUtil::ElementIsFloating(instruction->shape()) == in MatchesArCrsPattern()
56 ShapeUtil::ElementIsFloating(instruction->operand(0)->shape()); in MatchesArCrsPattern()
61 return ShapeUtil::ElementIsFloating(instruction->shape()); in MatchesArCrsPattern()
Dshape_inference.cc245 if (!ShapeUtil::ElementIsFloating(shape)) { in InferUnaryOpShape()
261 if (!ShapeUtil::ElementIsFloating(shape) && in InferUnaryOpShape()
273 } else if (ShapeUtil::ElementIsFloating(shape)) { in InferUnaryOpShape()
303 !ShapeUtil::ElementIsFloating(shape) && in InferUnaryOpShape()
332 if (!ShapeUtil::ElementIsFloating(shape)) { in InferUnaryOpShape()
463 if (!ShapeUtil::ElementIsFloating(operand_shape)) { in InferReducePrecisionShape()
955 if (!ShapeUtil::ElementIsFloating(lhs)) { in InferBinaryOpShape()
1219 if (!ShapeUtil::ElementIsFloating(operand_shape)) { in InferBatchNormTrainingShape()
1325 if (!ShapeUtil::ElementIsFloating(operand_shape)) { in InferBatchNormInferenceShape()
1471 if (!ShapeUtil::ElementIsFloating(operand_shape)) { in InferBatchNormGradShape()
[all …]
Dhlo_verifier.cc789 if (!ShapeUtil::ElementIsFloating(subshape)) { in CheckMixedPrecisionOperands()
Dhlo_evaluator_typed_visitor.h1770 if (ShapeUtil::ElementIsFloating(init_literals[0]->shape()) && in HandleReduce()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h258 if (ElementIsFloating(a) && ElementIsFloating(b)) { in SameElementTypeIgnoringFpPrecision()
441 static bool ElementIsFloating(const Shape& shape);
Dliteral_comparison.cc778 if (ShapeUtil::ElementIsFloating(expected.shape()) || in NearHelper()
Dshape_util.cc324 /* static */ bool ShapeUtil::ElementIsFloating(const Shape& shape) { in ElementIsFloating() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc48 return !ShapeUtil::ElementIsFloating(instruction.shape()) && in IsExpensive()
Dir_emitter.cc495 } else if (ShapeUtil::ElementIsFloating(lhs_shape)) { in HandleDot()
595 } else if (ShapeUtil::ElementIsFloating(lhs_shape)) { in HandleDot()
/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.cc1422 bool root_is_floating_point = ShapeUtil::ElementIsFloating(root_shape); in MatchReductionGenerator()