/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/module/ |
H A D | executor_group.py | 218 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 D | executor_group.py | 218 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 D | Executor.pm | 74 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 D | Symbol.pm | 888 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 D | Executor.pm | 74 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 D | Symbol.pm | 888 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 D | Group.pm | 284 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 D | Group.pm | 284 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 D | executor.jl | 109 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 D | executor.jl | 109 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 D | test_multi_device_exec.py | 23 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 D | test_subgraph_op.py | 132 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 D | test_multi_device_exec.py | 23 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 D | test_subgraph_op.py | 132 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 D | solver.py | 48 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 D | solver.py | 48 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 D | parameter.py | 105 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 D | parameter.py | 105 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 D | imperative.h | 56 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 D | imperative.h | 56 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 D | test_utils.py | 1209 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 D | test_utils.py | 1209 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 D | basic_layers.py | 345 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 D | basic_layers.py | 345 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 D | BasicLayers.pm | 439 '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',
|