| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_minimum_grad_op.py | 48 def gen_data(inputA_np, inputB_np, grad_=None): argument
57 if grad_ is None:
58 grad_ = Tensor(grad_)
62 output = net_me(inputA_me, inputB_me, Tensor(grad_))
72 grad_ = np.random.randn(1, 3, 2, 2).astype(np.float32)
73 output = gen_data(inputA_np, inputB_np, grad_)
89 grad_ = np.array([[[[0.42891738, 0.03434946, 0.06192983, 0.21216309, 0.37450036, 0.6619524],
160 output = gen_data(inputA, inputB, grad_)
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | minmax_grad.h | 58 if (grad_ == nullptr || idx_ < 0 || idx_ > 1 || node->func_graph() == nullptr) { in operator()
65 if (users.find(grad_) == users.end() || users[grad_].size() != 1) { in operator()
70 auto &inputs = grad_->inputs(); in operator()
94 void Visit(const CNodePtr &cnode) override { grad_ = cnode; } in Visit()
100 grad_ = nullptr; in Reset()
105 CNodePtr grad_{nullptr};
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_minimum_grad.py | 47 def gen_data(inputA_np, inputB_np, grad_=None): argument
56 if grad_ is None:
57 grad_ = np.random.randn(1, 3, 2, 2).astype(np.float32)
61 print(grad_)
|
| D | test_maximum_grad.py | 47 def gen_data(inputA_np, inputB_np, grad_=None): argument
54 if grad_ is None:
55 grad_ = np.random.randn(2).astype(np.float32)
63 output = net_me(inputA_me, inputB_me, Tensor(grad_))
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | base.py | 362 grad_ = GradOperation(self.get_all, self.get_by_list, self.sens_param)
371 return grad_(fn, weights)(*args)
375 return grad_(fn)(*args)
381 self._pynative_forward_run(grad_, args, kwargs, fn)
382 _pynative_executor.grad(grad_, fn, weights, *args, **kwargs)
387 grad_.pynative_ = True
391 return grad_(fn, weights)(*args, **kwargs)
394 return grad_(fn)(*args, **kwargs)
|
| /third_party/mindspore/tests/ut/cpp/kernel/cpu/ |
| D | sparse_apply_ftrl_cpu_kernel_test.cc | 39 grad_.clear(); in SetUp()
55 inputs_.push_back(CreateKernelAddress(grad_.data())); in CreateInputAddress()
70 std::vector<float> grad_; member in mindspore::kernel::SparseApplyFtrlCpuKernelTest
82 grad_.push_back(1.0); in TEST_F()
109 grad_.push_back(1.0); in TEST_F()
140 grad_.push_back(1.0); in TEST_F()
|
| D | sparse_apply_proximal_adagrad_cpu_kernel_test.cc | 35 grad_.clear(); in SetUp()
53 inputs_.push_back(CreateKernelAddress(grad_.data())); in CreateInputAddress()
67 std::vector<float> grad_; member in mindspore::kernel::SparseApplyProximalAdagradCpuKernelTest
81 grad_.push_back(1.0); in TEST_F()
107 grad_.push_back(1.0); in TEST_F()
137 grad_.push_back(1.0); in TEST_F()
|
| D | adam_delta_cpu_kernel_test.cc | 35 grad_.clear(); in SetUp()
57 inputs_.push_back(CreateKernelAddress(grad_.data(), elem_num_)); in CreateAddress()
64 std::vector<float> grad_; member in mindspore::kernel::AdamDeltaCpuKernelTest
83 grad_.push_back(1.0); in TEST_F()
|
| D | sparse_apply_lazy_adam_cpu_kernel_test.cc | 35 grad_.clear(); in SetUp()
57 inputs_.push_back(CreateKernelAddress(grad_.data())); in CreateInputAddress()
72 std::vector<float> grad_; member in mindspore::kernel::SparseApplyLazyAdamCpuKernelTest
90 grad_.push_back(1.0); in TEST_F()
117 grad_.push_back(1.0); in TEST_F()
148 grad_.push_back(1.0); in TEST_F()
|
| D | sparse_apply_adam_cpu_kernel_test.cc | 35 grad_.clear(); in SetUp()
57 inputs_.push_back(CreateKernelAddress(grad_.data())); in CreateInputAddress()
74 std::vector<float> grad_; member in mindspore::kernel::SparseApplyAdamCpuKernelTest
92 grad_.push_back(1.0); in TEST_F()
120 grad_.push_back(1.0); in TEST_F()
152 grad_.push_back(1.0); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | replace_momentum_cast_fusion.h | 30 grad_ = std::make_shared<Var>();
41 VarPtr grad_; variable
|
| D | replace_momentum_cast_fusion.cc | 25 VectorRef grad_cast = VectorRef({prim::kPrimCast, grad_}); in DefinePattern()
|