Home
last modified time | relevance | path

Searched refs:out_grads (Results 1 – 25 of 82) sorted by relevance

1234

/dports/misc/mxnet/incubator-mxnet-1.9.0/perl-package/AI-MXNet/lib/AI/MXNet/
H A DExecutor.pm226 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 DExecutor.pm226 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 Dexecutor.py156 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 Dexecutor.py156 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 Dtest_contrib_control_flow.py229 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 Dtest_subgraph.py131 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 Dtest_contrib_control_flow.py229 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 Dtest_subgraph.py131 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 Dsd_module.py160 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 Dsd_module.py160 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 Dmodule.py56 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 Dmodule.py56 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 Dmodel.py112 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 Dmodel.py112 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 Dautograd.py121 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 Dautograd.py121 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 Dsequential_module.py358 def backward(self, out_grads=None): argument
363 module.backward(out_grads=out_grads)
367 out_grads = module.get_input_grads()
H A Dpython_module.py316 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 Dsequential_module.py358 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 Dbase.py247 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 Dbase.py247 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 Dharvester.py60 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 Dtest_fusion.py53 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 Dtest_fusion.py53 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 Dimperative.h59 std::vector<NDArray> out_grads; variable
88 && info.out_grads.size() == 1; in IsVariable()

1234