Home
last modified time | relevance | path

Searched refs:get_imag (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_gpu_imag.cu.cc22 DEFINE_UNARY2(get_imag, complex64, complex128);
Dcwise_op_imag.cc24 UnaryOp<D##Device, functor::get_imag<C>>);
Dcwise_ops.h1115 struct get_imag
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc561 auto get_imag = [&](llvm::Value* x) { in EmitScalarDot() local
567 b_->CreateFMul(get_imag(lhs_value), get_imag(rhs_value))); in EmitScalarDot()
569 b_->CreateFMul(get_real(lhs_value), get_imag(rhs_value)), in EmitScalarDot()
570 b_->CreateFMul(get_imag(lhs_value), get_real(rhs_value))); in EmitScalarDot()