/dports/misc/mxnet/incubator-mxnet-1.9.0/perl-package/AI-MXNet/lib/AI/MXNet/ |
H A D | Executor.pm | 226 Maybe[AI::MXNet::NDArray|ArrayRef[AI::MXNet::NDArray]|HashRef[AI::MXNet::NDArray]] $out_grads=, 230 $out_grads //= []; 231 if(blessed $out_grads) 233 $out_grads = [$out_grads]; 235 elsif(ref $out_grads eq 'HASH') 237 $out_grads = [ @{ $out_grads }{ @{ $self->symbol->list_outputs() } } ]; 242 scalar(@{ $out_grads }), 243 [map { $_->handle } @{ $out_grads }],
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/perl-package/AI-MXNet/lib/AI/MXNet/ |
H A D | Executor.pm | 226 Maybe[AI::MXNet::NDArray|ArrayRef[AI::MXNet::NDArray]|HashRef[AI::MXNet::NDArray]] $out_grads=, 230 $out_grads //= []; 231 if(blessed $out_grads) 233 $out_grads = [$out_grads]; 235 elsif(ref $out_grads eq 'HASH') 237 $out_grads = [ @{ $out_grads }{ @{ $self->symbol->list_outputs() } } ]; 242 scalar(@{ $out_grads }), 243 [map { $_->handle } @{ $out_grads }],
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/ |
H A D | executor.py | 156 def backward(self, out_grads=None, is_train=True): argument 221 if out_grads is None: 222 out_grads = [] 223 elif isinstance(out_grads, NDArray): 224 out_grads = [out_grads] 225 elif isinstance(out_grads, dict): 226 out_grads = [out_grads[k] for k in self._symbol.list_outputs()] 228 for obj in out_grads: 231 handle_array = c_handle_array(out_grads) 234 mx_uint(len(out_grads)),
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/ |
H A D | executor.py | 156 def backward(self, out_grads=None, is_train=True): argument 221 if out_grads is None: 222 out_grads = [] 223 elif isinstance(out_grads, NDArray): 224 out_grads = [out_grads] 225 elif isinstance(out_grads, dict): 226 out_grads = [out_grads[k] for k in self._symbol.list_outputs()] 228 for obj in out_grads: 231 handle_array = c_handle_array(out_grads) 234 mx_uint(len(out_grads)),
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
H A D | test_contrib_control_flow.py | 229 executor.backward(out_grads=out_grads) 980 e_1.backward(out_grads) 983 e_2.backward(out_grads) 1026 out_grads = [] 1067 executor.backward(out_grads=out_grads) 1570 out_grads = [] 1573 e.backward(out_grads) 1584 res.backward(out_grads[0]) 1656 out_grads = [] 1664 e1.backward(out_grads) [all …]
|
H A D | test_subgraph.py | 131 out_grads = [mx.nd.random.uniform(-1, 1, shape=out.shape, ctx=default_context()) 133 e1.backward(out_grads) 134 e2.backward(out_grads)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
H A D | test_contrib_control_flow.py | 229 executor.backward(out_grads=out_grads) 980 e_1.backward(out_grads) 983 e_2.backward(out_grads) 1026 out_grads = [] 1067 executor.backward(out_grads=out_grads) 1570 out_grads = [] 1573 e.backward(out_grads) 1584 res.backward(out_grads[0]) 1656 out_grads = [] 1664 e1.backward(out_grads) [all …]
|
H A D | test_subgraph.py | 131 out_grads = [mx.nd.random.uniform(-1, 1, shape=out.shape, ctx=default_context()) 133 e1.backward(out_grads) 134 e2.backward(out_grads)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/stochastic-depth/ |
H A D | sd_module.py | 160 def backward(self, out_grads=None): argument 162 self._module_skip.backward(out_grads=out_grads) 165 self._input_grads = out_grads 168 self._module_compute.backward(out_grads=out_grads)
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/stochastic-depth/ |
H A D | sd_module.py | 160 def backward(self, out_grads=None): argument 162 self._module_skip.backward(out_grads=out_grads) 165 self._input_grads = out_grads 168 self._module_compute.backward(out_grads=out_grads)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/rnn/word_lm/ |
H A D | module.py | 56 def backward(self, out_grads=None): argument 59 self._module.backward(out_grads=out_grads)
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/rnn/word_lm/ |
H A D | module.py | 56 def backward(self, out_grads=None): argument 59 self._module.backward(out_grads=out_grads)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/reinforcement-learning/parallel_actor_critic/ |
H A D | model.py | 112 out_grads=[neg_advs, v_grads]) 119 def _forward_backward(self, data_batch, out_grads=None): argument 121 self.model.backward(out_grads=out_grads)
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/reinforcement-learning/parallel_actor_critic/ |
H A D | model.py | 112 out_grads=[neg_advs, v_grads]) 119 def _forward_backward(self, data_batch, out_grads=None): argument 121 self.model.backward(out_grads=out_grads)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/contrib/ |
H A D | autograd.py | 121 def backward(outputs, out_grads=None, retain_graph=False): argument 132 if out_grads is None: 141 for arr in out_grads:
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/contrib/ |
H A D | autograd.py | 121 def backward(outputs, out_grads=None, retain_graph=False): argument 132 if out_grads is None: 141 for arr in out_grads:
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/module/ |
H A D | sequential_module.py | 358 def backward(self, out_grads=None): argument 363 module.backward(out_grads=out_grads) 367 out_grads = module.get_input_grads()
|
H A D | python_module.py | 316 def backward(self, out_grads=None): argument 326 assert out_grads is None, 'For a loss module, out_grads should be None'
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/module/ |
H A D | sequential_module.py | 358 def backward(self, out_grads=None): argument 363 module.backward(out_grads=out_grads) 367 out_grads = module.get_input_grads()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/reinforcement-learning/dqn/ |
H A D | base.py | 247 def backward(self, out_grads=None, **arg_dict): argument 253 self.exe.backward(out_grads=out_grads) 255 def forward_backward(self, bucket_kwargs=None, out_grads=None, **arg_dict): argument 264 self.exe.backward(out_grads=out_grads)
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/reinforcement-learning/dqn/ |
H A D | base.py | 247 def backward(self, out_grads=None, **arg_dict): argument 253 self.exe.backward(out_grads=out_grads) 255 def forward_backward(self, bucket_kwargs=None, out_grads=None, **arg_dict): argument 264 self.exe.backward(out_grads=out_grads)
|
/dports/science/py-qcengine/qcengine-0.21.0/qcengine/procedures/nwchem_opt/ |
H A D | harvester.py | 60 out_psivars, out_mols, out_grads, version, module, error = harvest_output(nwout) 64 for qcvars, nwgrad, out_mol in zip(out_psivars, out_grads, out_mols):
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/gpu/ |
H A D | test_fusion.py | 53 out_grads = [mx.nd.ones_like(arr) for arr in fwd_orig] 54 orig_exec.backward(out_grads=out_grads) 56 fused_exec.backward(out_grads=out_grads)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/gpu/ |
H A D | test_fusion.py | 53 out_grads = [mx.nd.ones_like(arr) for arr in fwd_orig] 54 orig_exec.backward(out_grads=out_grads) 56 fused_exec.backward(out_grads=out_grads)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mxnet/ |
H A D | imperative.h | 59 std::vector<NDArray> out_grads; variable 88 && info.out_grads.size() == 1; in IsVariable()
|