Searched refs:ElementIsComplex (Results 1 – 13 of 13) sorted by relevance
25 !ShapeUtil::ElementIsComplex(inst->operand(0)->shape()); in InstructionMatchesPattern()
104 if (ShapeUtil::ElementIsComplex(a_shape)) { in CholeskyUnblocked()185 if (ShapeUtil::ElementIsComplex(a_shape)) { in BuildCholesky()
277 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()286 if (ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()297 if (ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()319 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()336 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()1956 if ((!ShapeUtil::ElementIsFloating(a) && !ShapeUtil::ElementIsComplex(a)) || in InferTriangularSolveShape()2007 if (!ShapeUtil::ElementIsFloating(a) && !ShapeUtil::ElementIsComplex(a)) { in InferCholeskyShape()
125 return ShapeUtil::ElementIsComplex(instruction.shape()); in IsExpensive()
256 } else if (ShapeUtil::ElementIsComplex(op->operand(0)->shape())) { in EmitUnaryOp()2559 ShapeUtil::ElementIsComplex(iota->shape()) in MakeElementGenerator()2588 if (ShapeUtil::ElementIsComplex(iota->shape())) { in MakeElementGenerator()
88 if (ShapeUtil::ElementIsComplex(operand->shape())) { in IsAnyOperandComplex()2809 !ShapeUtil::ElementIsComplex(abs_operand->shape())) { in HandleMultiply()3038 auto abs_a = ShapeUtil::ElementIsComplex(a->shape()) in HandleLog()
47 return !ShapeUtil::ElementIsComplex(hlo_shape) && in IsNonComplexNonBatchedMatrixVectorDot()
643 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in EmitNaiveLlvmIrGemm()710 if (ShapeUtil::ElementIsComplex(lhs_array_.GetShape())) { in EmitScalarDot()
1382 if (ShapeUtil::ElementIsComplex(root_shape)) { in MatchReductionGenerator()
489 static bool ElementIsComplex(const Shape& shape);
505 /* static */ bool ShapeUtil::ElementIsComplex(const Shape& shape) { in ElementIsComplex() function in xla::ShapeUtil558 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
818 ShapeUtil::ElementIsComplex(expected.shape())) { in NearHelper()
318 ShapeUtil::ElementIsComplex(expected.shape())) { in ComputeAndCompareLiteralWithStatus()