Searched refs:ElementIsIntegral (Results 1 – 10 of 10) sorted by relevance
294 if (!ShapeUtil::ElementIsIntegral(shape)) { in InferUnaryOpShape()302 if (!ShapeUtil::ElementIsIntegral(shape) && in InferUnaryOpShape()312 if (!ShapeUtil::ElementIsIntegral(shape)) { in InferUnaryOpShape()2411 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferDynamicSliceShape()2443 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicSliceShape()2517 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferDynamicUpdateSliceShape()2550 if (!ShapeUtil::ElementIsIntegral(first_index_shape)) { in InferDynamicUpdateSliceShape()3168 if (!ShapeUtil::ElementIsIntegral(start_indices_shape)) { in InferGatherShape()3405 if (!ShapeUtil::ElementIsIntegral(scatter_indices_shape)) { in InferScatterShape()
130 return !ShapeUtil::ElementIsIntegral(instruction.shape()) || in IsExpensive()
261 ShapeUtil::ElementIsIntegral(src_shape) && in IsConvertPairNoOp()262 ShapeUtil::ElementIsIntegral(intermediate_shape); in IsConvertPairNoOp()700 (ShapeUtil::ElementIsIntegral(add->shape()) || in HandleAdd()1029 if (ShapeUtil::ElementIsIntegral(divide->shape()) && in TryDivideToShift()1203 if (ShapeUtil::ElementIsIntegral(divide->shape())) { in HandleDivide()2936 if (ShapeUtil::ElementIsIntegral(remainder->shape()) && in TryRemainderToAnd()
236 if (ShapeUtil::ElementIsIntegral(op->operand(0)->shape()) || in EmitUnaryOp()806 if (ShapeUtil::ElementIsIntegral(op->operand(0)->shape()) || in EmitBinaryOp()
535 TF_RET_CHECK(ShapeUtil::ElementIsIntegral(lhs_shape)); in HandleDot()636 TF_RET_CHECK(ShapeUtil::ElementIsIntegral(lhs_shape)); in HandleDot()
322 /* static */ bool ShapeUtil::ElementIsIntegral(const Shape& shape) { in ElementIsIntegral() function in xla::ShapeUtil328 return ElementIsIntegral(shape) && ElementHasBitWidth(shape, bits); in ElementIsIntegralWithBits()
451 static bool ElementIsIntegral(const Shape& shape);
528 } else if (ShapeUtil::ElementIsIntegral(lhs_shape)) { in EmitNaiveLlvmIrGemm()
1611 bool root_is_integral = ShapeUtil::ElementIsIntegral(root_shape); in MatchReductionGenerator()
130 if (!ShapeUtil::ElementIsIntegral(*shape)) { in operator >>()