Home
last modified time | relevance | path

Searched refs:IsComplexType (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcast_op.cc53 } else if (xla::primitive_util::IsComplexType(src_type_) && in Compile()
54 !xla::primitive_util::IsComplexType(dst_type_)) { in Compile()
126 !xla::primitive_util::IsComplexType(src_type_) && in Compile()
127 !xla::primitive_util::IsComplexType(dst_type_), in Compile()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconstants.h37 primitive_util::IsComplexType(type))) { in ConstantR0WithType()
43 !primitive_util::IsComplexType(type)) { in ConstantR0WithType()
Dmath.cc1143 if (primitive_util::IsComplexType(shape.element_type())) { in Acos()
1200 if (primitive_util::IsComplexType(shape.element_type())) { in Acosh()
1237 if (primitive_util::IsComplexType(shape.element_type())) { in Asinh()
1275 if (primitive_util::IsComplexType(shape.element_type())) { in Atanh()
1322 if (primitive_util::IsComplexType(shape.element_type())) { in Sinh()
1352 primitive_util::IsComplexType(shape.element_type()) && conjugate; in MaybeConjugate()
/external/tensorflow/tensorflow/compiler/xla/
Dprimitive_util.h141 bool IsComplexType(PrimitiveType type);
Dprimitive_util.cc82 bool IsComplexType(PrimitiveType type) { return type == C64 || type == C128; } in IsComplexType() function
Dshape_util.cc478 return primitive_util::IsComplexType(shape.element_type()); in ElementIsComplex()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc454 if (primitive_util::IsComplexType(old_element_type) && in InferConvertShape()
455 !primitive_util::IsComplexType(new_element_type)) { in InferConvertShape()
478 if (primitive_util::IsComplexType(old_element_type) != in InferBitcastConvertShape()
479 primitive_util::IsComplexType(new_element_type)) { in InferBitcastConvertShape()
1890 if (!primitive_util::IsComplexType(in.element_type())) { in InferFftShape()
1924 if (!primitive_util::IsComplexType(in.element_type())) { in InferFftShape()
Delemental_ir_emitter.cc279 if (primitive_util::IsComplexType(to_type)) { in EmitIntegerUnaryOp()
386 if (primitive_util::IsComplexType(to_type)) { in EmitFloatUnaryOp()
523 primitive_util::IsComplexType(input_type) in EmitComplexUnaryOp()
554 TF_RET_CHECK(primitive_util::IsComplexType(from_type)); in EmitComplexUnaryOp()
556 TF_RET_CHECK(primitive_util::IsComplexType(to_type)); in EmitComplexUnaryOp()
822 } else if (primitive_util::IsComplexType(operand_type)) { in EmitBinaryOp()
2559 if (primitive_util::IsComplexType(primitive_type)) { in EmitMulAdd()
Dqr_expander.cc102 if (primitive_util::IsComplexType(type)) { in House()
Dhlo_verifier.cc624 !primitive_util::IsComplexType(primitive_type)) { in HandleIota()
Dhlo_parser.cc3198 if (!primitive_util::IsComplexType(shape.element_type())) { in ParseDenseLiteral()