Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc48 return !ShapeUtil::ElementIsComplex(hlo_shape) && in IsNonComplexMatrixVectorDot()
Ddot_op_emitter.cc495 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in EmitNaiveLlvmIrGemm()
556 if (ShapeUtil::ElementIsComplex(lhs_array_.GetShape())) { in EmitScalarDot()
Dir_emitter.cc1414 if (ShapeUtil::ElementIsComplex(root_shape)) { in MatchReductionGenerator()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc262 !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()
Dinstruction_fusion.cc113 return ShapeUtil::ElementIsComplex(instruction.shape()); in IsExpensive()
Delemental_ir_emitter.cc213 } 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()
Dalgebraic_simplifier.cc2366 !ShapeUtil::ElementIsComplex(lhs->operand(1)->shape()) && in HandlePower()
2367 !ShapeUtil::ElementIsComplex(rhs->shape()) && in HandlePower()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc490 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in HandleDot()
587 if (ShapeUtil::ElementIsComplex(lhs_shape)) { in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc320 /* static */ bool ShapeUtil::ElementIsComplex(const Shape& shape) { in ElementIsComplex() function in xla::ShapeUtil
373 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape); in ComplexComponentShape()
Dshape_util.h444 static bool ElementIsComplex(const Shape& shape);
Dliteral_comparison.cc779 ShapeUtil::ElementIsComplex(expected.shape())) { in NearHelper()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc318 ShapeUtil::ElementIsComplex(expected.shape())) { in ComputeAndCompareLiteralWithStatus()