| /third_party/mindspore/mindspore/core/ops/ |
| D | tensor_summary.cc | 32 …auto v_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape… in TensorSummaryInferShape() local
|
| D | scalar_summary.cc | 32 …auto v_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape… in ScalarSummaryInferShape() local
|
| D | apply_momentum.cc | 71 …auto v_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape… in ApplyMomentumInfer() local
|
| D | adam.cc | 33 …auto v_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->GetShapeT… in AdamInfer() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | sparse_apply_adam_ps_kernel.cc | 38 std::vector<size_t> &v_shape = *(shape_vec[2]); in InitKernel() local
|
| D | sparse_apply_lazy_adam_ps_kernel.cc | 39 std::vector<size_t> &v_shape = *(shape_vec[2]); 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
|
| D | math_ops.py | 1583 def infer_shape(self, x_shape, v_shape): argument
1652 def infer_shape(self, x_shape, v_shape): argument
|
| D | array_ops.py | 5545 def infer_shape(self, x_shape, v_shape): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | adam_weight_decay_gpu_kernel.h | 86 auto v_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in Init() local
|
| D | adam_gpu_kernel.h | 89 auto v_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | adam_delta_cpu_kernel.cc | 65 std::vector<size_t> v_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 1); in InitKernel() local
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 86 std::vector<size_t> v_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in InitKernel() local
|
| D | sparse_apply_adam_cpu_kernel.cc | 107 std::vector<size_t> v_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in InitKernel() local
|
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | step_auto_parallel_test.cc | 179 std::vector<Shapes> v_shape = {v_inputs_shape, v_outputs_shape}; in TEST_F() local
|