Home
last modified time | relevance | path

Searched defs:x_is_zero (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Digamma_utils.h367 T x_is_zero = (x == 0); in IgammaSingle() local
427 T x_is_zero = (x == 0); in IgammaGradASingle() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Drandom_gamma_grad_impl.cu139 bool x_is_zero = xSingle == 0; in GammaSingle() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Digamma_cpu_kernel.cc223 bool x_is_zero = (x == 0); in IgammaSingle() local
Digammagrada_cpu_kernel.cc260 bool x_is_zero = (x == 0); in IgammaGradASingle() local