Searched refs:IsComplexType (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | primitive_util.h | 142 bool IsComplexType(PrimitiveType type);
178 IsComplexType(type) ? ComplexComponentType(type) : type; in HigherPrecisionType()
181 IsComplexType(type), in HigherPrecisionType()
223 if (primitive_util::IsComplexType(to_type)) { in CastPreservesValues()
225 primitive_util::IsComplexType(from_type) in CastPreservesValues()
232 if (primitive_util::IsComplexType(from_type)) { in CastPreservesValues()
|
| D | primitive_util.cc | 83 bool IsComplexType(PrimitiveType type) { return type == C64 || type == C128; } in IsComplexType() function
|
| D | shape_util.cc | 540 return primitive_util::IsComplexType(shape.element_type()); in ElementIsComplex()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cast_op.cc | 53 } 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/ |
| D | self_adjoint_eig.cc | 52 !primitive_util::IsComplexType(type)) { in SelfAdjointEig()
75 primitive_util::IsComplexType(type) in SelfAdjointEig()
|
| D | constants.h | 37 primitive_util::IsComplexType(type))) { in ConstantR0WithType()
43 !primitive_util::IsComplexType(type)) { in ConstantR0WithType()
|
| D | math.cc | 1141 if (primitive_util::IsComplexType(shape.element_type())) { in Acos()
1198 if (primitive_util::IsComplexType(shape.element_type())) { in Acosh()
1235 if (primitive_util::IsComplexType(shape.element_type())) { in Asinh()
1273 if (primitive_util::IsComplexType(shape.element_type())) { in Atanh()
1320 if (primitive_util::IsComplexType(shape.element_type())) { in Sinh()
1350 primitive_util::IsComplexType(shape.element_type()) && conjugate; in MaybeConjugate()
|
| D | matrix.cc | 257 if (primitive_util::IsComplexType(shape.element_type())) { in Symmetrize()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | eigh_expander.cc | 103 bool is_complex = primitive_util::IsComplexType(w_tl_shape.element_type()); in HermitianEigenDecomposition2x2()
282 primitive_util::IsComplexType(shape.element_type()) in ComputeFrobeniusNorms()
442 !primitive_util::IsComplexType(type)) { in BuildEigh()
|
| D | elemental_ir_emitter.cc | 285 if (primitive_util::IsComplexType(to_type)) { in EmitIntegerUnaryOp()
392 if (primitive_util::IsComplexType(to_type)) { in EmitFloatUnaryOp()
568 primitive_util::IsComplexType(input_type) in EmitComplexUnaryOp()
594 TF_RET_CHECK(primitive_util::IsComplexType(from_type)); in EmitComplexUnaryOp()
596 TF_RET_CHECK(primitive_util::IsComplexType(to_type)); in EmitComplexUnaryOp()
863 } else if (primitive_util::IsComplexType(operand_type)) { in EmitBinaryOp()
2753 if (primitive_util::IsComplexType(primitive_type)) { in EmitMulAdd()
|
| D | qr_expander.cc | 130 if (primitive_util::IsComplexType(type)) { in House()
|
| D | shape_inference.cc | 451 if (primitive_util::IsComplexType(old_element_type) != in InferBitcastConvertShape()
452 primitive_util::IsComplexType(new_element_type)) { in InferBitcastConvertShape()
1891 if (!primitive_util::IsComplexType(in.element_type())) { in InferFftShape()
1925 if (!primitive_util::IsComplexType(in.element_type())) { in InferFftShape()
|
| D | hlo_graph_dumper.cc | 1239 if (primitive_util::IsComplexType(instr->shape().element_type())) { in ExtractGemmBackendConfigProps()
|
| D | hlo_verifier.cc | 987 !primitive_util::IsComplexType(primitive_type)) { in HandleIota()
|
| D | algebraic_simplifier.cc | 172 return !primitive_util::IsComplexType(hlo->shape().element_type()) && in IsScalarConstantNegInf()
178 return !primitive_util::IsComplexType(hlo->shape().element_type()) && in IsScalarConstantInf()
|
| D | hlo_parser.cc | 3654 if (!primitive_util::IsComplexType(shape.element_type())) { in ParseDenseLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_mhlo_to_scalar_op.h | 263 struct IsComplexType {
289 IsFloatType, ScalarFOpOrVoid, IsComplexType,
306 return MapMhloOpToScalarOpImpl<IsComplexType, ::mlir::complex::AbsOp>{}(
894 return MapMhloOpToScalarOpImpl<IsFloatType, arith::DivFOp, IsComplexType,
950 IsComplexType, ::mlir::complex::NegOp>{}(
992 return MapMhloOpToScalarOpImpl<IsFloatType, math::PowFOp, IsComplexType,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_utils.cc | 197 bool is_complex = primitive_util::IsComplexType(element_type); in ToNcclDataTypeAndCountMultiplier()
|
| D | nccl_all_reduce_thunk.cc | 193 } else if (primitive_util::IsComplexType(type)) { in MatchAllReduceComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2207 if (primitive_util::IsComplexType(operand_shape->element_type()) && in ConvertElementType()
2208 !primitive_util::IsComplexType(new_element_type)) { in ConvertElementType()
|