| /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
265 validator.check("var_shape", var_shape, "v_shape", v_shape, Rel.EQ, self.name)
267 return m_shape, v_shape, m_shape
441 def infer_shape(self, d_shape, s_shape, v_shape, a_shape, l_shape, g_shape, m_shape): argument
442 return v_shape
543 def infer_shape(self, var_shape, m_shape, v_shape, lr_shape, beta1_shape, beta2_shape, argument
546 validator.check("var_shape", var_shape, "v_shape", v_shape, Rel.EQ, self.name)
548 return var_shape, m_shape, v_shape
|
| D | debug_ops.py | 143 v_shape = value['shape']
145 if len(v_shape) != image_dim:
242 v_shape = value['shape']
244 if not v_shape:
|
| D | math_ops.py | 1583 def infer_shape(self, x_shape, v_shape): argument
1584 validator.check("x", len(x_shape), "v", len(v_shape), Rel.EQ, self.name)
1585 validator.check("size of indices", len(self.indices), "v's first dimension", v_shape[0],
1593 … validator.check('v dim %d' % idx, v_shape[idx], "x dim %d" % idx, x_shape[idx], Rel.EQ, self.name)
1652 def infer_shape(self, x_shape, v_shape): argument
1653 validator.check("x", len(x_shape), "v", len(v_shape), Rel.EQ, self.name)
1654 validator.check("size of indices", len(self.indices), "v's first dimension", v_shape[0],
1662 … validator.check('v dim %d' % idx, v_shape[idx], "x dim %d" % idx, x_shape[idx], Rel.EQ, self.name)
|
| D | nn_ops.py | 2667 def infer_shape(self, v_shape, a_shape, l_shape, g_shape, m_shape): argument
2668 return v_shape
4595 … def infer_shape(self, var_shape, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
4598 validator.check("var_shape", var_shape, "v_shape", v_shape, Rel.EQ, self.name)
4600 return var_shape, m_shape, v_shape
4702 def infer_shape(self, var_shape, m_shape, v_shape, lr_shape, beta1_shape, beta2_shape, argument
4705 validator.check("var_shape", var_shape, "v_shape", v_shape, Rel.EQ, self.name)
4707 return var_shape, m_shape, v_shape
4815 def infer_shape(self, m_shape, v_shape, beta1_power_shape, beta2_power_shape, lr_shape, argument
4818 validator.check("grad_shape", grad_shape, "v_shape", v_shape, Rel.EQ, self.name)
[all …]
|
| D | array_ops.py | 5545 def infer_shape(self, x_shape, v_shape): argument
5546 validator.check("x", len(x_shape), "v", len(v_shape), Rel.EQ, self.name)
5547 validator.check("size of indices", len(self.indices), "v's first dimension", v_shape[0],
5555 … validator.check('v dim %d' % idx, v_shape[idx], "x dim %d" % idx, x_shape[idx], Rel.EQ, self.name)
|
| /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
88 …ut_ = CHECK_NULL_INPUT(variable_shape) || CHECK_NULL_INPUT(m_shape) || CHECK_NULL_INPUT(v_shape) || in Init()
103 for (size_t i = 0; i < v_shape.size(); i++) { in Init()
104 v_size_ *= v_shape[i]; in Init()
|
| D | adam_gpu_kernel.h | 89 auto v_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in Init() local
91 …ut_ = CHECK_NULL_INPUT(variable_shape) || CHECK_NULL_INPUT(m_shape) || CHECK_NULL_INPUT(v_shape) || in Init()
106 for (size_t i = 0; i < v_shape.size(); i++) { in Init()
107 v_size_ *= v_shape[i]; in Init()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | adam.cc | 33 …auto v_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->GetShapeT… in AdamInfer() local
36 CheckAndConvertUtils::Check("var_shape", var_shape, kEqual, "v_shape", v_shape, prim_name); in AdamInfer()
50 auto output2 = std::make_shared<abstract::AbstractTensor>(infer_v_type, v_shape); in AdamInfer()
|
| D | tensor_summary.cc | 32 …auto v_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape… in TensorSummaryInferShape() local
33 …(void)CheckAndConvertUtils::CheckInteger("v rank", int64_t(v_shape.size()), kGreaterEqual, 1, prim… in TensorSummaryInferShape()
|
| D | scalar_summary.cc | 32 …auto v_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape… in ScalarSummaryInferShape() local
33 …(void)CheckAndConvertUtils::CheckInteger("v rank", int64_t(v_shape.size()), kLessEqual, 1, prim_na… in ScalarSummaryInferShape()
|
| D | apply_momentum.cc | 71 …auto v_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape… in ApplyMomentumInfer() local
89 return std::make_shared<abstract::AbstractTensor>(element, v_shape); in ApplyMomentumInfer()
|
| /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
44 Shard(&v_shape, 0); in InitKernel()
51 if (!IsSameShape(var_shape, v_shape)) { in InitKernel()
|
| D | sparse_apply_lazy_adam_ps_kernel.cc | 39 std::vector<size_t> &v_shape = *(shape_vec[2]); in InitKernel() local
45 Shard(&v_shape, 0); in InitKernel()
56 if (!IsSameShape(var_shape, v_shape)) { in InitKernel()
|
| /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
71 if (!IsSameShape(delta_shape, v_shape)) { in InitKernel()
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 86 std::vector<size_t> v_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in InitKernel() local
95 if (!IsSameShape(var_shape, v_shape)) { in InitKernel()
|
| D | sparse_apply_adam_cpu_kernel.cc | 107 std::vector<size_t> v_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in InitKernel() local
116 if (!IsSameShape(var_shape, v_shape)) { in InitKernel()
|
| /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
181 std::shared_ptr<OperatorInfo> v_matmul_info = NewOperatorInstance(v_prim, v_attrs, v_shape); in TEST_F()
|