Searched refs:ElementIsIntegral (Results 1 – 9 of 9) sorted by relevance
320 if (!ShapeUtil::ElementIsIntegral(shape)) { in InferUnaryOpShape()328 if (!ShapeUtil::ElementIsIntegral(shape) && in InferUnaryOpShape()338 if (!ShapeUtil::ElementIsIntegral(shape)) { in InferUnaryOpShape()2503 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferDynamicSliceShape()2535 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicSliceShape()2609 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferDynamicUpdateSliceShape()2642 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicUpdateSliceShape()3311 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferGatherShape()3548 if (!ShapeUtil::ElementIsIntegral(scatter_indices_shape)) { in InferScatterShape()
132 return !ShapeUtil::ElementIsIntegral(instruction.shape()) || in IsExpensive()
263 ShapeUtil::ElementIsIntegral(src_shape) && in IsConvertPairNoOp()264 ShapeUtil::ElementIsIntegral(intermediate_shape); in IsConvertPairNoOp()916 (ShapeUtil::ElementIsIntegral(add->shape()) || in HandleAdd()1467 if (ShapeUtil::ElementIsIntegral(divide->shape()) && in TryDivideToShift()1641 if (ShapeUtil::ElementIsIntegral(divide->shape())) { in HandleDivide()3184 ShapeUtil::ElementIsIntegral(x->shape())) { in HandleCompare()3691 if (ShapeUtil::ElementIsIntegral(remainder->shape()) && in TryRemainderToAnd()4394 ShapeUtil::ElementIsIntegral(operand->shape()); in HandleDynamicSlice()
238 if (ShapeUtil::ElementIsIntegral(op->operand(0)->shape()) || in EmitUnaryOp()817 if (ShapeUtil::ElementIsIntegral(op->operand(0)->shape()) || in EmitBinaryOp()
500 static bool ElementIsIntegral(const Shape& shape);
432 /* static */ bool ShapeUtil::ElementIsIntegral(const Shape& shape) { in ElementIsIntegral() function in xla::ShapeUtil438 return ElementIsIntegral(shape) && ElementHasBitWidth(shape, bits); in ElementIsIntegralWithBits()
651 } else if (ShapeUtil::ElementIsIntegral(lhs_shape)) { in EmitNaiveLlvmIrGemm()
1388 bool root_is_integral = ShapeUtil::ElementIsIntegral(root_shape); in MatchReductionGenerator()
205 if (!ShapeUtil::ElementIsIntegral(*shape)) { in operator >>()