Home
last modified time | relevance | path

Searched defs:Grad (Results 1 – 25 of 132) sorted by relevance

123456

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dpdist_grad.cc55 struct Grad { struct
56 static inline T abs(T x) { return static_cast<T>(std::abs(*((float *)&x))); } in abs()
58 static inline T pow(T x, float p) { return static_cast<T>(std::pow(*((float *)&x), p)); } in pow()
60 static inline T sign(T x) { return x > T{0.0f} ? T{1.0f} : T{-1.0f}; } in sign()
62 struct o_grad {
66 struct t_grad {
77 struct p_grad {
83 struct i_grad {
90 …tatic uint32_t ParallelForPdistGrad(T *grad, T *x, T *dist, T *y, float p, CpuKernelContext &ctx) { in ParallelForPdistGrad()
117 …ine uint32_t PdistGradComputeKernel(T *grad, T *x, T *dist, T *y, float p, CpuKernelContext &ctx) { in PdistGradComputeKernel()
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/
Dtest_env_ops.py90 class Grad(nn.Cell): class
148 class Grad(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_avgpool_grad_op.py27 class Grad(nn.Cell): class
Dtest_pad_v3_grad_op.py22 import mindspore.ops.operations._grad_ops as Grad namespace
Dtest_hsigmoid_op.py29 class Grad(nn.Cell): class
Dtest_abs_op.py28 class Grad(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_tbe_ops/
Dtest_bn_infer_grad.py28 class Grad(nn.Cell): class
Dtest_sigmoid_grad.py37 class Grad(nn.Cell): class
Dtest_relu_grad.py27 class Grad(nn.Cell): class
Dtest_resize_nearest_neighbor_grad.py37 class Grad(nn.Cell): class
Dtest_maxpool_grad.py27 class Grad(nn.Cell): class
Dtest_batchnorm_grad.py30 class Grad(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_dynamic_gru_v2_grad.py42 class Grad(nn.Cell): class
Dtest_dense_grad.py26 class Grad(nn.Cell): class
Dtest_maxpool_grad.py27 class Grad(nn.Cell): class
Dtest_pad_v3_grad.py23 import mindspore.ops.operations._grad_ops as Grad namespace
Dtest_batchnorm_grad.py30 class Grad(nn.Cell): class
Dtest_simplemean_grad.py27 class Grad(nn.Cell): class
Dtest_conv_grad.py29 class Grad(nn.Cell): class
Dtest_maxpool_with_argmax_grad.py27 class Grad(nn.Cell): class
Dtest_maxpool_grad_with_argmax_v2.py28 class Grad(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_acl_ops/
Dtest_relu.py28 class Grad(nn.Cell): class
Dtest_reduce.py28 class Grad(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/
Dtest_dynamic_shape_softplus_grad.py33 class Grad(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/ut/python/nn/
Dtest_nn_pad.py34 class Grad(nn.Cell): class

123456