Searched defs:linear_shape (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | sparse_apply_ftrl_ps_kernel.cc | 36 std::vector<size_t> linear_shape = *(shape_vec[2]); in InitKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | ftrl_gpu_kernel.h | 83 auto linear_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in Init() local
|
| D | sparse_ftrl_gpu_kernel.h | 83 auto linear_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_ftrl_cpu_kernel.cc | 90 std::vector<size_t> linear_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in InitKernel() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 5223 def infer_shape(self, var_shape, accum_shape, linear_shape, grad_shape, indices_shape): argument
7014 …def infer_shape(self, var_shape, accum_shape, linear_shape, grad_shape, lr_shape, l1_shape, l2_sha… argument
7124 def check_shape(self, var_shape, accum_shape, linear_shape, grad_shape, indices_shape): argument
7232 def infer_shape(self, var_shape, accum_shape, linear_shape, grad_shape, indices_shape): argument
|