Home
last modified time | relevance | path

Searched defs:num_group (Results 1 – 25 of 111) sorted by relevance

12345

/dports/misc/mxnet/incubator-mxnet-1.9.0/example/image-classification/symbols/
H A Dresnext.py28 def residual_unit(data, num_filter, stride, dim_match, name, bottle_neck=True, num_group=32, bn_mom… argument
101 def resnext(units, num_stages, filter_list, num_classes, num_group, image_shape, bottle_neck=True, … argument
157 def get_symbol(num_classes, num_layers, image_shape, num_group=32, conv_workspace=256, dtype='float… argument
H A Dmobilenet.py38 def Conv(data, num_filter=1, kernel=(1, 1), stride=(1, 1), pad=(0, 0), num_group=1, name='', suffix… argument
H A Dmobilenetv2.py39 def mobilenet_unit(data, num_filter=1, kernel=(1, 1), stride=(1, 1), pad=(0, 0), num_group=1, if_ac… argument
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/image-classification/symbols/
H A Dresnext.py28 def residual_unit(data, num_filter, stride, dim_match, name, bottle_neck=True, num_group=32, bn_mom… argument
101 def resnext(units, num_stages, filter_list, num_classes, num_group, image_shape, bottle_neck=True, … argument
157 def get_symbol(num_classes, num_layers, image_shape, num_group=32, conv_workspace=256, dtype='float… argument
H A Dmobilenet.py38 def Conv(data, num_filter=1, kernel=(1, 1), stride=(1, 1), pad=(0, 0), num_group=1, name='', suffix… argument
H A Dmobilenetv2.py39 def mobilenet_unit(data, num_filter=1, kernel=(1, 1), stride=(1, 1), pad=(0, 0), num_group=1, if_ac… argument
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/
H A Dmkldnn_deconvolution-inl.h65 const uint32_t num_group) { in IOLogicalSwapDesc()
74 inline void IOLogicalSwapMKLDNNMem(const NDArray &arr, const uint32_t num_group) { in IOLogicalSwapMKLDNNMem()
90 inline mkldnn::memory::desc GetDeconvWeightsDesc(const NDArray &weights, const uint32_t num_group) { in GetDeconvWeightsDesc()
152 inline const mkldnn::memory *MKLDNNDeconvFwd::WeightsMem(const uint32_t num_group, in WeightsMem()
255 inline void MKLDNNDeconvBwd::IOSwapWeightsTensors(const uint32_t num_group, in IOSwapWeightsTensors()
271 inline const mkldnn::memory *MKLDNNDeconvBwd::WeightsMem(const uint32_t num_group, in WeightsMem()
292 inline mkldnn_output_t MKLDNNDeconvBwd::WeightsGradMem(const uint32_t num_group, in WeightsGradMem()
H A Dmkldnn_deconvolution.cc97 void MKLDNNDeconvFwd::ControlWeightsFormat(const uint32_t num_group, const bool is_train, in ControlWeightsFormat()
120 void MKLDNNDeconvFwd::Execute(const uint32_t num_group, const OpReqType req, in Execute()
250 void MKLDNNDeconvBwd::Execute(const uint32_t num_group, const std::vector<OpReqType> &req, in Execute()
264 const mkldnn::memory *MKLDNNDeconvBwd::ScheduleBwdData(const uint32_t num_group, in ScheduleBwdData()
285 void MKLDNNDeconvBwd::ScheduleBwdWeights(const uint32_t num_group, in ScheduleBwdWeights()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/
H A Dmkldnn_deconvolution-inl.h65 const uint32_t num_group) { in IOLogicalSwapDesc()
74 inline void IOLogicalSwapMKLDNNMem(const NDArray &arr, const uint32_t num_group) { in IOLogicalSwapMKLDNNMem()
90 inline mkldnn::memory::desc GetDeconvWeightsDesc(const NDArray &weights, const uint32_t num_group) { in GetDeconvWeightsDesc()
152 inline const mkldnn::memory *MKLDNNDeconvFwd::WeightsMem(const uint32_t num_group, in WeightsMem()
255 inline void MKLDNNDeconvBwd::IOSwapWeightsTensors(const uint32_t num_group, in IOSwapWeightsTensors()
271 inline const mkldnn::memory *MKLDNNDeconvBwd::WeightsMem(const uint32_t num_group, in WeightsMem()
292 inline mkldnn_output_t MKLDNNDeconvBwd::WeightsGradMem(const uint32_t num_group, in WeightsGradMem()
H A Dmkldnn_deconvolution.cc97 void MKLDNNDeconvFwd::ControlWeightsFormat(const uint32_t num_group, const bool is_train, in ControlWeightsFormat()
120 void MKLDNNDeconvFwd::Execute(const uint32_t num_group, const OpReqType req, in Execute()
250 void MKLDNNDeconvBwd::Execute(const uint32_t num_group, const std::vector<OpReqType> &req, in Execute()
264 const mkldnn::memory *MKLDNNDeconvBwd::ScheduleBwdData(const uint32_t num_group, in ScheduleBwdData()
285 void MKLDNNDeconvBwd::ScheduleBwdWeights(const uint32_t num_group, in ScheduleBwdWeights()
/dports/misc/py-xgboost/xgboost-1.5.1/src/linear/
H A Dcoordinate_common.h81 inline std::pair<double, double> GetGradient(int group_idx, int num_group, int fidx, in GetGradient()
111 inline std::pair<double, double> GetGradientParallel(int group_idx, int num_group, int fidx, in GetGradientParallel()
145 inline std::pair<double, double> GetBiasGradientParallel(int group_idx, int num_group, in GetBiasGradientParallel()
175 inline void UpdateResidualParallel(int fidx, int group_idx, int num_group, in UpdateResidualParallel()
206 inline void UpdateBiasResidualParallel(int group_idx, int num_group, float dbias, in UpdateBiasResidualParallel()
H A Dupdater_gpu_coordinate.cu173 GradientPair GetBiasGradient(int group_idx, int num_group) { in GetBiasGradient()
197 GradientPair GetGradient(int group_idx, int num_group, int fidx) { in GetGradient()
/dports/misc/xgboost/xgboost-1.5.1/src/linear/
H A Dcoordinate_common.h81 inline std::pair<double, double> GetGradient(int group_idx, int num_group, int fidx, in GetGradient()
111 inline std::pair<double, double> GetGradientParallel(int group_idx, int num_group, int fidx, in GetGradientParallel()
145 inline std::pair<double, double> GetBiasGradientParallel(int group_idx, int num_group, in GetBiasGradientParallel()
175 inline void UpdateResidualParallel(int fidx, int group_idx, int num_group, in UpdateResidualParallel()
206 inline void UpdateBiasResidualParallel(int group_idx, int num_group, float dbias, in UpdateBiasResidualParallel()
H A Dupdater_gpu_coordinate.cu173 GradientPair GetBiasGradient(int group_idx, int num_group) { in GetBiasGradient()
197 GradientPair GetGradient(int group_idx, int num_group, int fidx) { in GetGradient()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A Dconvolution-inl.h63 uint32_t num_group; member
H A Ddeconvolution-inl.h58 uint32_t num_group; member
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/
H A Dconvolution-inl.h63 uint32_t num_group; member
H A Ddeconvolution-inl.h58 uint32_t num_group; member
/dports/science/py-OpenMC/openmc-0.12.2/src/
H A Dmgxs.cpp275 hid_t xs_id, const vector<double>& temperature, int num_group, int num_delay) in Mgxs()
311 int num_group, int num_delay) in Mgxs()
/dports/science/openmc/openmc-0.12.2/src/
H A Dmgxs.cpp275 hid_t xs_id, const vector<double>& temperature, int num_group, int num_delay) in Mgxs()
311 int num_group, int num_delay) in Mgxs()
/dports/graphics/realsr-ncnn-vulkan/realsr-ncnn-vulkan-20210210/src/ncnn/tools/mxnet/
H A Dmxnet2ncnn.cpp1285 int num_group = n.attr("num_group"); in main() local
1305 int num_group = n.attr("num_group"); in main() local
1886 int num_group = n.attr("num_group"); in main() local
1984 int num_group = n.attr("num_group"); in main() local
/dports/misc/py-xgboost/xgboost-1.5.1/plugin/updater_oneapi/
H A Dpredictor_oneapi.cc160 int num_group; member in xgboost::predictor::DeviceModelOneAPI
307 int num_group = model.learner_model_param->num_output_group; in DevicePredictInternal() local
/dports/misc/xgboost/xgboost-1.5.1/plugin/updater_oneapi/
H A Dpredictor_oneapi.cc160 int num_group; member in xgboost::predictor::DeviceModelOneAPI
307 int num_group = model.learner_model_param->num_output_group; in DevicePredictInternal() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/model_zoo/vision/
H A Dmobilenet.py49 num_group=1, active=True, relu6=False): argument
/dports/misc/py-gluoncv/gluon-cv-0.9.0/gluoncv/model_zoo/
H A Dmobilenet.py48 num_group=1, active=True, relu6=False, norm_layer=BatchNorm, norm_kwargs=None): argument

12345