Searched refs:ElementIsComplex (Results 1 – 12 of 12) sorted by relevance
47 return !ShapeUtil::ElementIsComplex(hlo_shape) && in IsNonComplexNonBatchedMatrixVectorDot()
515 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in EmitNaiveLlvmIrGemm()579 if (ShapeUtil::ElementIsComplex(lhs_array_.GetShape())) { in EmitScalarDot()
1602 if (ShapeUtil::ElementIsComplex(root_shape)) { in MatchReductionGenerator()
262 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()271 if (ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()282 if (ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()304 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()321 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()1931 if ((!ShapeUtil::ElementIsFloating(a) && !ShapeUtil::ElementIsComplex(a)) || in InferTriangularSolveShape()1982 if (!ShapeUtil::ElementIsFloating(a) && !ShapeUtil::ElementIsComplex(a)) { in InferCholeskyShape()
123 return ShapeUtil::ElementIsComplex(instruction.shape()); in IsExpensive()
85 if (ShapeUtil::ElementIsComplex(operand->shape())) { in IsAnyOperandComplex()1910 ShapeUtil::ElementIsComplex(dot->shape())) && in HandleDot()1972 ShapeUtil::ElementIsComplex(dot->shape()) in HandleDot()2350 auto abs_a = ShapeUtil::ElementIsComplex(a->shape()) in HandleLog()2835 !ShapeUtil::ElementIsComplex(lhs->operand(1)->shape()) && in HandlePower()2836 !ShapeUtil::ElementIsComplex(rhs->shape()) && in HandlePower()
239 } else if (ShapeUtil::ElementIsComplex(op->operand(0)->shape())) { in EmitUnaryOp()2274 ShapeUtil::ElementIsComplex(iota->shape()) in MakeElementGenerator()2303 if (ShapeUtil::ElementIsComplex(iota->shape())) { in MakeElementGenerator()
527 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in HandleDot()624 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in HandleDot()
367 /* static */ bool ShapeUtil::ElementIsComplex(const Shape& shape) { in ElementIsComplex() function in xla::ShapeUtil420 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
457 static bool ElementIsComplex(const Shape& shape);
825 ShapeUtil::ElementIsComplex(expected.shape())) { in NearHelper()
318 ShapeUtil::ElementIsComplex(expected.shape())) { in ComputeAndCompareLiteralWithStatus()