Searched refs:l2_shape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | apply_adagrad_d_a.cc | 43 …auto l2_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex6]->BuildSha… in InferShape() local
51 …(void)CheckAndConvertUtils::CheckInteger("l2_shape size", SizeToInt(l2_shape.size()), kEqual, inpu… in InferShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 88 std::vector<size_t> l2_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 4); in InitKernel() local
120 if (!l2_shape.empty()) { in InitKernel()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 5329 def infer_shape(self, var_shape, accum_shape, lr_shape, l1_shape, l2_shape, argument
6283 def infer_shape(self, var_shape, accum_shape, lr_shape, l1_shape, l2_shape, grad_shape): argument
6294 l2_shp_len = len(l2_shape)
6297 validator.check_int(l2_shape[0], 1, Rel.EQ, "l2_shape[0]", self.name)
6407 def check_shape(self, var_shape, accum_shape, lr_shape, l1_shape, l2_shape, argument
6822 def infer_shape(self, var_shape, alpha_shape, l1_shape, l2_shape, delta_shape): argument
6832 l2_shape_len = len(l2_shape)
6835 validator.check_int(l2_shape[0], 1, Rel.EQ, "l2_shape[0]", self.name)
7014 …f infer_shape(self, var_shape, accum_shape, linear_shape, grad_shape, lr_shape, l1_shape, l2_shape, argument
|