Searched refs:ElementIsComplex (Results 1 – 12 of 12) sorted by relevance
43 if (ShapeUtil::ElementIsComplex(input_shape) || in PotentiallyImplementedAsEigenConvolution()44 ShapeUtil::ElementIsComplex(kernel_shape)) { in PotentiallyImplementedAsEigenConvolution()
822 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in Emit()884 if (ShapeUtil::ElementIsComplex(lhs_array_.GetShape())) { in EmitScalarDot()
1337 if (ShapeUtil::ElementIsComplex(root_shape)) { in MatchReductionGenerator()
267 ShapeUtil::ElementIsComplex(expected.shape())) { in ComputeAndCompareLiteralWithStatus()324 ShapeUtil::ElementIsComplex(expected.shape())); in ComputeAndCompareLiteralWithStatus()
713 ShapeUtil::ElementIsComplex(expected.shape())) { in Near()
88 return ShapeUtil::ElementIsComplex(instruction.shape()); in IsExpensive()
334 !ShapeUtil::ElementIsComplex(arg)) { in InferUnaryOpShape()343 if (!ShapeUtil::ElementIsComplex(arg)) { in InferUnaryOpShape()351 if (ShapeUtil::ElementIsComplex(arg)) { in InferUnaryOpShape()
1400 !ShapeUtil::ElementIsComplex(lhs->operand(1)->shape()) && in HandlePower()1401 !ShapeUtil::ElementIsComplex(rhs->shape()) && in HandlePower()
209 } else if (ShapeUtil::ElementIsComplex(op->operand(0)->shape())) { in EmitUnaryOp()
458 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in HandleDot()550 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in HandleDot()
370 static bool ElementIsComplex(const Shape& shape);
312 /* static */ bool ShapeUtil::ElementIsComplex(const Shape& shape) { in ElementIsComplex() function in xla::ShapeUtil361 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()