Searched refs:ElementIsIntegral (Results 1 – 7 of 7) sorted by relevance
364 static bool ElementIsIntegral(const Shape& shape);
269 /* static */ bool ShapeUtil::ElementIsIntegral(const Shape& shape) { in ElementIsIntegral() function in xla::ShapeUtil275 return ElementIsIntegral(shape) && ElementHasBitWidth(shape, bits); in ElementIsIntegralWithBits()
270 TF_RET_CHECK(ShapeUtil::ElementIsIntegral(expected.shape()) || in ComputeAndCompareLiteralWithStatus()
2030 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferDynamicSliceShape()2093 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferDynamicUpdateSliceShape()2559 if (!ShapeUtil::ElementIsIntegral(gather_indices_shape)) { in InferGatherShape()
206 } else if (ShapeUtil::ElementIsIntegral(op->operand(0)->shape()) || in EmitUnaryOp()663 if (ShapeUtil::ElementIsIntegral(op->operand(0)->shape()) || in EmitBinaryOp()
598 if (ShapeUtil::ElementIsIntegral(divide->shape())) { in HandleDivide()
1346 bool root_is_integral = ShapeUtil::ElementIsIntegral(root_shape); in MatchReductionGenerator()