Home
last modified time | relevance | path

Searched refs:ElementIsComplex (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcholesky_expander.cc104 if (ShapeUtil::ElementIsComplex(a_shape)) { in CholeskyUnblocked()
185 if (ShapeUtil::ElementIsComplex(a_shape)) { in BuildCholesky()
Dshape_inference.cc288 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()
297 if (ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()
308 if (ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()
330 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()
347 !ShapeUtil::ElementIsComplex(shape)) { in InferUnaryOpShape()
1964 if ((!ShapeUtil::ElementIsFloating(a) && !ShapeUtil::ElementIsComplex(a)) || in InferTriangularSolveShape()
2015 if (!ShapeUtil::ElementIsFloating(a) && !ShapeUtil::ElementIsComplex(a)) { in InferCholeskyShape()
Dinstruction_fusion.cc125 return ShapeUtil::ElementIsComplex(instruction.shape()); in IsExpensive()
Delemental_ir_emitter.cc241 } else if (ShapeUtil::ElementIsComplex(op->operand(0)->shape())) { in EmitUnaryOp()
2373 ShapeUtil::ElementIsComplex(iota->shape()) in MakeElementGenerator()
2402 if (ShapeUtil::ElementIsComplex(iota->shape())) { in MakeElementGenerator()
Dalgebraic_simplifier.cc87 if (ShapeUtil::ElementIsComplex(operand->shape())) { in IsAnyOperandComplex()
2719 !ShapeUtil::ElementIsComplex(abs_operand->shape())) { in HandleMultiply()
2948 auto abs_a = ShapeUtil::ElementIsComplex(a->shape()) in HandleLog()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc47 return !ShapeUtil::ElementIsComplex(hlo_shape) && in IsNonComplexNonBatchedMatrixVectorDot()
Ddot_op_emitter.cc638 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in EmitNaiveLlvmIrGemm()
705 if (ShapeUtil::ElementIsComplex(lhs_array_.GetShape())) { in EmitScalarDot()
Dir_emitter.cc1379 if (ShapeUtil::ElementIsComplex(root_shape)) { in MatchReductionGenerator()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h506 static bool ElementIsComplex(const Shape& shape);
Dshape_util.cc477 /* static */ bool ShapeUtil::ElementIsComplex(const Shape& shape) { in ElementIsComplex() function in xla::ShapeUtil
530 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
Dliteral_comparison.cc814 ShapeUtil::ElementIsComplex(expected.shape())) { in NearHelper()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc318 ShapeUtil::ElementIsComplex(expected.shape())) { in ComputeAndCompareLiteralWithStatus()