Home
last modified time | relevance | path

Searched refs:grad_req (Results 1 – 25 of 170) sorted by relevance

1234567

/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/module/
H A Dexecutor_group.py218 grad_req = 'null'
226 if isinstance(grad_req, str):
227 self.grad_req = {}
230 self.grad_req[k] = 'null' if k in self.fixed_param_names else grad_req
232 self.grad_req[k] = grad_req if self.inputs_need_grad else 'null'
234 self.grad_req[k] = 'null'
237 self.grad_req = dict(zip(self.arg_names, grad_req))
238 elif isinstance(grad_req, dict):
239 self.grad_req = {}
247 self.grad_req.update(grad_req)
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/module/
H A Dexecutor_group.py218 grad_req = 'null'
226 if isinstance(grad_req, str):
227 self.grad_req = {}
230 self.grad_req[k] = 'null' if k in self.fixed_param_names else grad_req
232 self.grad_req[k] = grad_req if self.inputs_need_grad else 'null'
234 self.grad_req[k] = 'null'
237 self.grad_req = dict(zip(self.arg_names, grad_req))
238 elif isinstance(grad_req, dict):
239 self.grad_req = {}
247 self.grad_req.update(grad_req)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/perl-package/AI-MXNet/lib/AI/MXNet/
H A DExecutor.pm74 my ($symbol, $ctx, $grad_req, $group2ctx)
79 if(ref $grad_req)
81 if(ref $grad_req eq 'ARRAY')
83 $grad_req = [ @{ $grad_req }];
85 elsif(ref $grad_req eq 'HASH')
87 $grad_req = { %{ $grad_req } };
97 $self->_grad_req($grad_req);
499 grad_req => $self->_grad_req,
H A DSymbol.pm888 if(defined $grad_req)
890 if(not ref $grad_req)
894 elsif(ref $grad_req eq 'ARRAY')
900 elsif(ref $grad_req eq 'HASH')
1005 grad_req => $grad_req,
1088 $grad_req //= 'write';
1114 if(not ref $grad_req)
1117 unless exists $req_map->{ $grad_req };
1120 elsif(ref $grad_req eq 'ARRAY')
1128 if(exists $grad_req->{ $name })
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/perl-package/AI-MXNet/lib/AI/MXNet/
H A DExecutor.pm74 my ($symbol, $ctx, $grad_req, $group2ctx)
79 if(ref $grad_req)
81 if(ref $grad_req eq 'ARRAY')
83 $grad_req = [ @{ $grad_req }];
85 elsif(ref $grad_req eq 'HASH')
87 $grad_req = { %{ $grad_req } };
97 $self->_grad_req($grad_req);
499 grad_req => $self->_grad_req,
H A DSymbol.pm888 if(defined $grad_req)
890 if(not ref $grad_req)
894 elsif(ref $grad_req eq 'ARRAY')
900 elsif(ref $grad_req eq 'HASH')
1005 grad_req => $grad_req,
1088 $grad_req //= 'write';
1114 if(not ref $grad_req)
1117 unless exists $req_map->{ $grad_req };
1120 elsif(ref $grad_req eq 'ARRAY')
1128 if(exists $grad_req->{ $name })
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/perl-package/AI-MXNet/lib/AI/MXNet/Executor/
H A DGroup.pm284 my %grad_req;
285 if(not ref $self->grad_req)
291 $grad_req{$k} = exists $fixed_param_names{ $k } ? 'null' : $self->grad_req;
295 $grad_req{$k} = $self->inputs_need_grad ? $self->grad_req : 'null';
299 $grad_req{$k} = 'null';
303 elsif(ref $self->grad_req eq 'ARRAY')
305 @grad_req{ @{ $self->_p->arg_names } } = @{ $self->grad_req };
321 $grad_req{$k} = 'null';
324 %grad_req = (%grad_req, %{ $self->grad_req });
326 $self->grad_req(\%grad_req);
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/perl-package/AI-MXNet/lib/AI/MXNet/Executor/
H A DGroup.pm284 my %grad_req;
285 if(not ref $self->grad_req)
291 $grad_req{$k} = exists $fixed_param_names{ $k } ? 'null' : $self->grad_req;
295 $grad_req{$k} = $self->inputs_need_grad ? $self->grad_req : 'null';
299 $grad_req{$k} = 'null';
303 elsif(ref $self->grad_req eq 'ARRAY')
305 @grad_req{ @{ $self->_p->arg_names } } = @{ $self->grad_req };
321 $grad_req{$k} = 'null';
324 %grad_req = (%grad_req, %{ $self->grad_req });
326 $self->grad_req(\%grad_req);
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/julia/src/
H A Dexecutor.jl109 grad_req = GRAD_WRITE) argument
117 if isa(grad_req, GRAD_REQ)
118 reqs = MX_uint[MX_uint(grad_req) for i=1:length(args)]
119 elseif isa(grad_req, Vector{GRAD_REQ})
120 @assert(length(grad_req) == length(args))
121 reqs = MX_uint[MX_uint.(grad_req)...]
122 elseif isa(grad_req, Dict{Symbol, GRAD_REQ})
155 if grad_req != GRAD_NOP
163 # if isa(grad_req, Dict{Symbol, GRAD_REQ})
164 # shapes = filter(x -> grad_req[x[1]] != GRAD_NOP,shapes)
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/julia/src/
H A Dexecutor.jl109 grad_req = GRAD_WRITE) argument
117 if isa(grad_req, GRAD_REQ)
118 reqs = MX_uint[MX_uint(grad_req) for i=1:length(args)]
119 elseif isa(grad_req, Vector{GRAD_REQ})
120 @assert(length(grad_req) == length(args))
121 reqs = MX_uint[MX_uint.(grad_req)...]
122 elseif isa(grad_req, Dict{Symbol, GRAD_REQ})
155 if grad_req != GRAD_NOP
163 # if isa(grad_req, Dict{Symbol, GRAD_REQ})
164 # shapes = filter(x -> grad_req[x[1]] != GRAD_NOP,shapes)
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/
H A Dtest_multi_device_exec.py23 def check_ctx_group(group2ctx, grad_req, mlp, set_stage1): argument
26 data=(1,200), grad_req=grad_req)
60 for grad_req in grad_reqs:
61 check_ctx_group(group2ctx, grad_req, mlp, set_stage1)
H A Dtest_subgraph_op.py132 exe = sym.simple_bind(ctx=mx.current_context(), grad_req='null')
133 partitioned_exe = partitioned_sym.simple_bind(ctx=mx.current_context(), grad_req='null')
158 exe = sym.simple_bind(ctx=mx.current_context(), grad_req='null')
199 exe = sym.bind(ctx=mx.current_context(), args=arg_array, aux_states=aux_array, grad_req='null')
201 aux_states=aux_array, grad_req='null')
223 grad_req='null')
261 exe1 = sym.simple_bind(ctx=mx.current_context(), grad_req='null')
272 exe2 = part_sym.simple_bind(ctx=mx.current_context(), grad_req='null')
287 exe1 = sym.simple_bind(ctx=mx.current_context(), grad_req='null')
298 exe2 = part_sym.simple_bind(ctx=mx.current_context(), grad_req='null')
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/
H A Dtest_multi_device_exec.py23 def check_ctx_group(group2ctx, grad_req, mlp, set_stage1): argument
26 data=(1,200), grad_req=grad_req)
60 for grad_req in grad_reqs:
61 check_ctx_group(group2ctx, grad_req, mlp, set_stage1)
H A Dtest_subgraph_op.py132 exe = sym.simple_bind(ctx=mx.current_context(), grad_req='null')
133 partitioned_exe = partitioned_sym.simple_bind(ctx=mx.current_context(), grad_req='null')
158 exe = sym.simple_bind(ctx=mx.current_context(), grad_req='null')
199 exe = sym.bind(ctx=mx.current_context(), args=arg_array, aux_states=aux_array, grad_req='null')
201 aux_states=aux_array, grad_req='null')
223 grad_req='null')
261 exe1 = sym.simple_bind(ctx=mx.current_context(), grad_req='null')
272 exe2 = part_sym.simple_bind(ctx=mx.current_context(), grad_req='null')
287 exe1 = sym.simple_bind(ctx=mx.current_context(), grad_req='null')
298 exe2 = part_sym.simple_bind(ctx=mx.current_context(), grad_req='null')
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/fcn-xs/
H A Dsolver.py48 grad_req='write', argument
58 if grad_req != 'null':
87 grad_req=grad_req,
130 grad_req=grad_req,
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/fcn-xs/
H A Dsolver.py48 grad_req='write', argument
58 if grad_req != 'null':
87 grad_req=grad_req,
130 grad_req=grad_req,
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/
H A Dparameter.py105 def __init__(self, name, grad_req='write', shape=None, dtype=mx_real_t, argument
125 self.grad_req = grad_req
141 def grad_req(self): member in Parameter
144 @grad_req.setter
145 def grad_req(self, req): member in Parameter
370 if self.grad_req == 'null':
385 self._grad, self.grad_req)
704 def grad_req(self): member in Constant
707 @grad_req.setter
708 def grad_req(self, req): member in Constant
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/
H A Dparameter.py105 def __init__(self, name, grad_req='write', shape=None, dtype=mx_real_t, argument
125 self.grad_req = grad_req
141 def grad_req(self): member in Parameter
144 @grad_req.setter
145 def grad_req(self, req): member in Parameter
370 if self.grad_req == 'null':
385 self._grad, self.grad_req)
704 def grad_req(self): member in Constant
707 @grad_req.setter
708 def grad_req(self, req): member in Constant
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mxnet/
H A Dimperative.h56 OpReqType grad_req; variable
63 grad_req(kNullOp), fresh_out_grad(false) {} in AGInfo()
68 if (info.grad_req != kNullOp) return; in Clear()
87 return info.grad_req != kNullOp && info.outputs.size() == 1 in IsVariable()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/include/mxnet/
H A Dimperative.h56 OpReqType grad_req; variable
63 grad_req(kNullOp), fresh_out_grad(false) {} in AGInfo()
68 if (info.grad_req != kNullOp) return; in Clear()
87 return info.grad_req != kNullOp && info.outputs.size() == 1 in IsVariable()
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/
H A Dtest_utils.py1209 executor = out.bind(ctx, grad_req=grad_req,
1423 grad_req = {k:grad_req for k in sym.list_arguments()}
1425 grad_req = {k:v for k, v in zip(sym.list_arguments(), grad_req)}
1428 aux_states=aux_states, grad_req=grad_req)
1499 if grad_req is None:
1500 grad_req = 'write'
1502 exe = sym.simple_bind(grad_req=grad_req, ctx=ctx, **kwargs)
1507 exe = sym.simple_bind(grad_req=grad_req, ctx=ctx,
1624 exe_list.append(s.simple_bind(grad_req=grad_req, **ctx))
1661 if grad_req != 'null':
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/
H A Dtest_utils.py1209 executor = out.bind(ctx, grad_req=grad_req,
1423 grad_req = {k:grad_req for k in sym.list_arguments()}
1425 grad_req = {k:v for k, v in zip(sym.list_arguments(), grad_req)}
1428 aux_states=aux_states, grad_req=grad_req)
1499 if grad_req is None:
1500 grad_req = 'write'
1502 exe = sym.simple_bind(grad_req=grad_req, ctx=ctx, **kwargs)
1507 exe = sym.simple_bind(grad_req=grad_req, ctx=ctx,
1624 exe_list.append(s.simple_bind(grad_req=grad_req, **ctx))
1661 if grad_req != 'null':
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/nn/
H A Dbasic_layers.py345 self.gamma = self.params.get('gamma', grad_req='write' if scale else 'null',
349 self.beta = self.params.get('beta', grad_req='write' if center else 'null',
353 self.running_mean = self.params.get('running_mean', grad_req='null',
358 self.running_var = self.params.get('running_var', grad_req='null',
655 self.gamma = self.params.get('gamma', grad_req='write' if scale else 'null',
658 self.beta = self.params.get('beta', grad_req='write' if center else 'null',
743 self.gamma = self.params.get('gamma', grad_req='write' if scale else 'null',
746 self.beta = self.params.get('beta', grad_req='write' if center else 'null',
834 self.gamma = self.params.get('gamma', grad_req='write' if scale else 'null',
837 self.beta = self.params.get('beta', grad_req='write' if center else 'null',
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/nn/
H A Dbasic_layers.py345 self.gamma = self.params.get('gamma', grad_req='write' if scale else 'null',
349 self.beta = self.params.get('beta', grad_req='write' if center else 'null',
353 self.running_mean = self.params.get('running_mean', grad_req='null',
358 self.running_var = self.params.get('running_var', grad_req='null',
655 self.gamma = self.params.get('gamma', grad_req='write' if scale else 'null',
658 self.beta = self.params.get('beta', grad_req='write' if center else 'null',
743 self.gamma = self.params.get('gamma', grad_req='write' if scale else 'null',
746 self.beta = self.params.get('beta', grad_req='write' if center else 'null',
834 self.gamma = self.params.get('gamma', grad_req='write' if scale else 'null',
837 self.beta = self.params.get('beta', grad_req='write' if center else 'null',
/dports/misc/mxnet/incubator-mxnet-1.9.0/perl-package/AI-MXNet/lib/AI/MXNet/Gluon/NN/
H A DBasicLayers.pm439 'gamma', grad_req => $self->scale ? 'write' : 'null',
446 'beta', grad_req => $self->center ? 'write' : 'null',
453 'running_mean', grad_req => 'null',
460 'running_var', grad_req => $self->center ? 'write' : 'null',
670 'gamma', grad_req => $self->scale ? 'write' :'null',
677 'beta', grad_req => $self->scale ? 'write' :'null',
782 'gamma', grad_req => $self->scale ? 'write' :'null',
789 'beta', grad_req => $self->scale ? 'write' :'null',

1234567