Searched defs:gradient_shape (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | adagrad_gpu_kernel.h | 84 auto gradient_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 3); in Init() local
|
| D | momentum_gpu_kernel.h | 73 auto gradient_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 3); in Init() local
|
| D | ftrl_gpu_kernel.h | 84 auto gradient_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 3); in Init() local
|
| D | sparse_apply_proximal_adagrad_kernel.h | 83 auto gradient_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 5); in Init() local
|
| D | adam_weight_decay_gpu_kernel.h | 87 auto gradient_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 8); in Init() local
|
| D | adam_gpu_kernel.h | 90 auto gradient_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 9); in Init() local
|
| D | sparse_ftrl_gpu_kernel.h | 84 auto gradient_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 3); in Init() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | ctcloss.cc | 79 abstract::ShapePtr gradient_shape; in InferShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | update_thor_gradient.h | 191 auto gradient_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in SetProperty() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 3026 def check_shape(self, parameters_shape, gradient_shape, learning_rate_shape, argument
6910 …def infer_shape(self, weight_shape, gradient_shape, norm_weight_shape, norm_gradient_shape, weight… argument
|