Searched refs:accum_shape (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | sparse_apply_ftrl_ps_kernel.cc | 35 std::vector<size_t> accum_shape = *(shape_vec[1]); in InitKernel() local
41 Shard(&accum_shape, 0); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 85 std::vector<size_t> accum_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in InitKernel() local
94 if (!IsSameShape(var_shape, accum_shape)) { in InitKernel()
|
| D | sparse_apply_ftrl_cpu_kernel.cc | 89 std::vector<size_t> accum_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in InitKernel() local
96 if (!IsSameShape(var_shape, accum_shape)) { in InitKernel()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 3027 accum_shape, momentum_shape, stat_shape): argument
3031 validator.check_positive_int(len(accum_shape), "accumulation rank", self.name)
5223 def infer_shape(self, var_shape, accum_shape, linear_shape, grad_shape, indices_shape): argument
5224 validator.check('var shape', var_shape, 'accum shape', accum_shape, Rel.EQ, self.name)
5329 def infer_shape(self, var_shape, accum_shape, lr_shape, l1_shape, l2_shape, argument
5764 def infer_shape(self, var_shape, accum_shape, accum_update_shape, lr_shape, rho_shape, argument
5766 validator.check("accum_shape", accum_shape, "var_shape", var_shape, Rel.EQ, self.name)
5781 return var_shape, accum_shape, accum_update_shape
5871 def infer_shape(self, var_shape, accum_shape, lr_shape, grad_shape): argument
5872 validator.check('accum shape', accum_shape, 'var shape', var_shape, Rel.EQ, self.name)
[all …]
|