Searched defs:SqrtGrad (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | sqrt_grad.h | 28 SqrtGrad() : PrimitiveC(kNameSqrtGrad) { InitIOName({"out_backprop", "input"}, {"output"}); } in SqrtGrad() function
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | sqrt_grad.py | 20 class SqrtGrad(Expander): class
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | eltwise_grad_cpu_kernel.cc | 77 void EltWiseGradCPUKernel<T>::SqrtGrad(const T *input1, const T *input2, T *out, size_t start, size… in SqrtGrad() function in mindspore::kernel::EltWiseGradCPUKernel
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | unary_op_grad_impl.cu | 126 void SqrtGrad(const T *input, const T *dout, T *output, const size_t count, cudaStream_t cuda_strea… in SqrtGrad() function
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 160 class SqrtGrad(PrimitiveWithInfer): class
|