Home
last modified time | relevance | path

Searched defs:ComplexAbs (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dcomplex_abs.h34 ComplexAbs() : BaseOperator(kNameComplexAbs) { InitIOName({"x"}, {"y"}); } in ComplexAbs() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dcomplex_abs_impl.cu101 cudaError_t ComplexAbs(const size_t input_elements, const T *x0, S *y, const uint32_t &device_id, c… in ComplexAbs() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dmath_ops_proto.cc47 IMPLEMT_INFERFUNC(ComplexAbs, ComplexAbsInfer) { in IMPLEMT_INFERFUNC() argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dmath_ops.py3626 class ComplexAbs(Primitive): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini685 [ComplexAbs] section
686 opInfo.subTypeOfInferShape=1
687 opInfo.opsFlag=OPS_FLAG_CLOSE
688 opInfo.engine=DNN_VM_AICPU
689 opInfo.flagPartial=False
690 opInfo.computeCost=100
691 opInfo.flagAsync=False
692 opInfo.opKernelLib=CUSTAICPUKernel
693 opInfo.formatAgnostic=False
694 opInfo.kernelSo=libcust_cpu_kernels.so
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/audio/kernels/
Daudio_utils.cc106 Status ComplexAbs(const std::shared_ptr<Tensor> &input, std::shared_ptr<Tensor> *output) { in ComplexAbs() function