Searched refs:full_param_ (Results 1 – 6 of 6) sorted by relevance
48 full_param_(nnvm::get<MKLDNNFCFullParam>(attrs.parsed)) {} in SgMKLDNNFCOp()68 MKLDNNFCFullParam full_param_; member in mxnet::op::SgMKLDNNFCOp95 auto &mkldnn_param = full_param_.mkldnn_param; in Forward()96 auto &default_param = full_param_.default_param; in Forward()290 full_param_.eltwise_param.scale = in Forward()303 full_param_.output_scales.resize(num_channel); in Forward()306 full_param_.output_scales[i] = tmp_scale_ / weight_scales_[i]; in Forward()309 full_param_.output_scales.resize(1); in Forward()310 full_param_.output_scales[0] = tmp_scale_ / weight_scales_[0]; in Forward()323 full_param_.output_scales.resize(0); in Forward()[all …]
638 const size_t num_fusion = full_param_.layer_params.size(); in Init()641 for (auto& layer_param : full_param_.layer_params) { in Init()647 for (auto& layer_param : full_param_.layer_params) { in Init()656 for (auto& layer_param : full_param_.layer_params) { in Init()667 const RNNParam &default_param = full_param_.default_param; in Init()959 const RNNParam& default_param = full_param_.default_param; in Forward()1079 const RNNParam& default_param = full_param_.default_param; in Backward()1125 if (full_param_.default_param.mode == rnn_enum::kLstm in Backward()1151 auto desc = mkldnn::memory::desc(full_param_.layer_params.back().src_dims, in Backward()1180 if (full_param_.default_param.mode == rnn_enum::kLstm) in Backward()[all …]
408 full_param_(MKLDNNRnnFullParamParser(param, seq_len, batch_size, input_size)) { } in MKLDNNRnnOp()420 const RNNParam& GetParam() const { return full_param_.default_param; } in GetParam()425 MKLDNNRnnFullParam full_param_; variable