Searched refs:l1_shape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | apply_adagrad_d_a.cc | 42 …auto l1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex5]->BuildSha… in InferShape() local
49 …(void)CheckAndConvertUtils::CheckInteger("l1_shape size", SizeToInt(l1_shape.size()), kEqual, inpu… in InferShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 87 std::vector<size_t> l1_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 3); in InitKernel() local
117 if (!l1_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
6290 l1_shp_len = len(l1_shape)
6293 validator.check_int(l1_shape[0], 1, Rel.EQ, "l1_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
6828 l1_shape_len = len(l1_shape)
6831 validator.check_int(l1_shape[0], 1, Rel.EQ, "l1_shape[0]", self.name)
7014 …def infer_shape(self, var_shape, accum_shape, linear_shape, grad_shape, lr_shape, l1_shape, l2_sha… argument
|