| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | softmax_cross_entropy_infer.c | 36 TensorC *grads = outputs[1]; in SoftmaxCrossEntropyInferShape() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | softmax_cross_entropy_with_logits.cc | 32 …ropyWithLogitsCPUKernel::ForwardPostExecute(const float *labels, const float *logits, float *grads, in ForwardPostExecute()
67 float *grads = nullptr; in Execute() local
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_resize_nearest_neighbor_grad.py | 44 def construct(self, images, grads): argument
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_nn_pad.py | 41 def construct(self, x, grads): argument
|
| /third_party/mindspore/mindspore/boost/ |
| D | grad_accumulation.py | 62 def construct(self, loss, grads): argument
|
| D | boost_cell_wrapper.py | 226 def gradient_accumulation_process(self, loss, grads): argument
244 def adasum_process(self, loss, grads): argument
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | ada_grad.py | 156 def construct(self, grads): argument
|
| D | proximal_ada_grad.py | 183 def construct(self, grads): argument
|
| D | ftrl.py | 222 def construct(self, grads): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | ctcloss_gpu_kernel.h | 236 …tForWS(int *label_value_pcr, int *cum_labels_length, int *label_squence_length, T *costs, T *grads, in MemsetForWS()
299 T *grads; variable
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_ms_function.py | 65 def construct(self, grads): argument
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_momentum.py | 67 def construct(self, grads): argument
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | primitive_py.cc | 93 py::tuple grads; in check_bprop_out() local
202 py::tuple grads = check_bprop_out(grads_obj, py_args); in RunCellBpropFunction() local
|
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | grad_reducer.py | 396 def construct(self, grads): argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | cell_wrapper.py | 120 grads, argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | ctcloss_impl.cu | 172 … int *label_squence_length, int *cum_labels_length, T *cost, T *grads, T *prob_num, in CTCLossKernel()
422 … T *cost, T *grads, T *prob_num, bool ignore_longer_outputs_than_inputs, cudaStream_t stream) { in CTCLoss()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 701 def infer_shape(self, grads, x, diff_scale, diff_offset, scale, batch_mean, batch_variance): argument
704 def infer_dtype(self, grads, x, diff_scale, diff_offset, scale, batch_mean, batch_variance): argument
716 def infer_shape(self, grads, x, batch_mean, batch_variance): argument
719 def infer_dtype(self, grads, x, batch_mean, batch_variance): argument
1689 def __infer__(self, grads, size): argument
2196 def infer_dtype(self, grads, x, y): argument
2201 def infer_shape(self, grads, x, y): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 505 std::vector<AnfNodePtr> grads; in GenerateFuncGraph() local
546 std::vector<AnfNodePtr> grads; in GenerateFuncGraph() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | kpynative.cc | 948 std::vector<AnfNodePtr> grads; in BuildMakeSequenceBprop() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | worker.cc | 570 const float *original_data, const float *grads, int *indices, in BuildSparseValue()
|