Searched refs:ElementIsComplex (Results 1 – 12 of 12) sorted by relevance
48 return !ShapeUtil::ElementIsComplex(hlo_shape) && in IsNonComplexMatrixVectorDot()
495 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in EmitNaiveLlvmIrGemm()556 if (ShapeUtil::ElementIsComplex(lhs_array_.GetShape())) { in EmitScalarDot()
1414 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()313 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()1905 if ((!ShapeUtil::ElementIsFloating(a) && !ShapeUtil::ElementIsComplex(a)) || in InferTriangularSolveShape()1956 if (!ShapeUtil::ElementIsFloating(a) && !ShapeUtil::ElementIsComplex(a)) { in InferCholeskyShape()
113 return ShapeUtil::ElementIsComplex(instruction.shape()); in IsExpensive()
213 } else if (ShapeUtil::ElementIsComplex(op->operand(0)->shape())) { in EmitUnaryOp()2329 ShapeUtil::ElementIsComplex(iota->shape()) in MakeElementGenerator()2359 if (ShapeUtil::ElementIsComplex(iota->shape())) { in MakeElementGenerator()
2366 !ShapeUtil::ElementIsComplex(lhs->operand(1)->shape()) && in HandlePower()2367 !ShapeUtil::ElementIsComplex(rhs->shape()) && in HandlePower()
490 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in HandleDot()587 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in HandleDot()
320 /* static */ bool ShapeUtil::ElementIsComplex(const Shape& shape) { in ElementIsComplex() function in xla::ShapeUtil373 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
444 static bool ElementIsComplex(const Shape& shape);
779 ShapeUtil::ElementIsComplex(expected.shape())) { in NearHelper()
318 ShapeUtil::ElementIsComplex(expected.shape())) { in ComputeAndCompareLiteralWithStatus()