Searched defs:learning_rate_shape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | sparse_apply_proximal_adagrad_kernel.h | 82 auto learning_rate_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in Init() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 3026 def check_shape(self, parameters_shape, gradient_shape, learning_rate_shape, argument
3128 …def infer_shape(self, var_shape, mean_square_shape, moment_shape, learning_rate_shape, grad_shape,… argument
3246 learning_rate_shape, decay_shape, momentum_shape, epsilon_shape): argument
6911 learning_rate_shape): argument
|