Searched refs:ElementIsIntegral (Results 1 – 8 of 8) sorted by relevance
294 if (!ShapeUtil::ElementIsIntegral(shape)) { in InferUnaryOpShape()302 if (!ShapeUtil::ElementIsIntegral(shape) && in InferUnaryOpShape()2295 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferDynamicSliceShape()2327 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicSliceShape()2401 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferDynamicUpdateSliceShape()2434 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicUpdateSliceShape()2966 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferGatherShape()3173 if (!ShapeUtil::ElementIsIntegral(scatter_indices_shape)) { in InferScatterShape()
554 (ShapeUtil::ElementIsIntegral(add->shape()) || in HandleAdd()862 if (ShapeUtil::ElementIsIntegral(divide->shape()) && in TryDivideToShift()1053 if (ShapeUtil::ElementIsIntegral(divide->shape())) { in HandleDivide()2464 if (ShapeUtil::ElementIsIntegral(remainder->shape()) && in TryRemainderToAnd()
210 if (ShapeUtil::ElementIsIntegral(op->operand(0)->shape()) || in EmitUnaryOp()687 if (ShapeUtil::ElementIsIntegral(op->operand(0)->shape()) || in EmitBinaryOp()
498 TF_RET_CHECK(ShapeUtil::ElementIsIntegral(lhs_shape)); in HandleDot()599 TF_RET_CHECK(ShapeUtil::ElementIsIntegral(lhs_shape)); in HandleDot()
275 /* static */ bool ShapeUtil::ElementIsIntegral(const Shape& shape) { in ElementIsIntegral() function in xla::ShapeUtil281 return ElementIsIntegral(shape) && ElementHasBitWidth(shape, bits); in ElementIsIntegralWithBits()
438 static bool ElementIsIntegral(const Shape& shape);
1423 bool root_is_integral = ShapeUtil::ElementIsIntegral(root_shape); in MatchReductionGenerator()
91 if (!ShapeUtil::ElementIsIntegral(shape)) { in operator >>()