| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | _basic.py | 43 def _check_mn_shape(m_shape, n_shape, km_shape, kn_shape): argument
69 def _check_bias(shape_bias, shape_a, shape_b, m_shape, n_shape): argument
|
| D | matmul_cube_fracz_left_cast_impl.py | 54 def _get_block(trans_a, trans_b, m_shape, n_shape, km_shape, kn_shape): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | sparse_apply_lazy_adam_ps_kernel.cc | 38 std::vector<size_t> &m_shape = *(shape_vec[1]); in InitKernel() local
|
| D | sparse_apply_adam_ps_kernel.cc | 37 std::vector<size_t> &m_shape = *(shape_vec[1]); in InitKernel() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | inner_ops.py | 262 def infer_shape(self, grad_shape, v_shape, m_shape, var_shape, beta1_shape, sub1_shape, argument
441 def infer_shape(self, d_shape, s_shape, v_shape, a_shape, l_shape, g_shape, m_shape): argument
543 def infer_shape(self, var_shape, m_shape, v_shape, lr_shape, beta1_shape, beta2_shape, argument
|
| D | nn_ops.py | 2667 def infer_shape(self, v_shape, a_shape, l_shape, g_shape, m_shape): argument
4595 … def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
4702 def infer_shape(self, var_shape, m_shape, v_shape, lr_shape, beta1_shape, beta2_shape, argument
4815 def infer_shape(self, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
4955 … def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
5106 … def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
5631 def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, lr_shape, argument
6512 def infer_shape(self, var_shape, m_shape, lr_shape, alpha_shape, sign_decay_shape, argument
6639 def infer_shape(self, var_shape, m_shape, lr_shape, logbase_shape, sign_decay_shape, argument
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | adam.cc | 32 …auto m_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetShapeT… in AdamInfer() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | adam_weight_decay_gpu_kernel.h | 85 auto m_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in Init() local
|
| D | adam_gpu_kernel.h | 88 auto m_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | adam_delta_cpu_kernel.cc | 64 std::vector<size_t> m_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); in InitKernel() local
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 85 std::vector<size_t> m_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in InitKernel() local
|
| D | sparse_apply_adam_cpu_kernel.cc | 106 std::vector<size_t> m_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in InitKernel() local
|
| /third_party/boost/boost/math/distributions/ |
| D | inverse_gamma.hpp | 116 RealType m_shape; // distribution shape member in boost::math::inverse_gamma_distribution
|
| D | gamma.hpp | 97 RealType m_shape; // distribution shape member in boost::math::gamma_distribution
|
| D | pareto.hpp | 158 RealType m_shape; // distribution shape (k) or alpha member in boost::math::pareto_distribution
|
| D | weibull.hpp | 96 RealType m_shape; // distribution shape member in boost::math::weibull_distribution
|