Searched refs:Imag (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | unary_op_complex_gpu_kernel.cc | 25 MS_REG_GPU_KERNEL_ONE(Imag, KernelAttr().AddInputAttr(kNumberTypeComplex64).AddOutputAttr(kNumberTy…
27 MS_REG_GPU_KERNEL_ONE(Imag, KernelAttr().AddInputAttr(kNumberTypeComplex128).AddOutputAttr(kNumberT…
|
| D | unary_op_complex_gpu_kernel.h | 54 Imag(input_addr, output_addr, inputs[0]->size / sizeof(Complex<T>), in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | unary_op_impl.cuh | 71 void Imag(const Complex<T> *input, T *output, const size_t count, cudaStream_t cuda_stream);
|
| D | unary_op_impl.cu | 519 void Imag(const Complex<T> *input, T *output, const size_t count, cudaStream_t cuda_stream) { in Imag() function
725 template void Imag<float>(const Complex<float> *input, float *output, const size_t count, cudaStrea…
731 template void Imag<double>(const Complex<double> *input, double *output, const size_t count, cudaSt…
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 62 MatrixInverse, IndexAdd, Erfinv, Conj, Real, Imag)
|
| D | math_ops.py | 5428 class Imag(PrimitiveWithInfer): class
|
| /third_party/skia/third_party/externals/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Utils/ |
| D | SimplifyLibCalls.cpp | 1375 Value *Real, *Imag; in optimizeCAbs() local
1380 Imag = B.CreateExtractValue(Op, 1, "imag"); in optimizeCAbs()
1384 Imag = CI->getArgOperand(1); in optimizeCAbs()
1388 Value *ImagImag = B.CreateFMul(Imag, Imag); in optimizeCAbs()
|