Home
last modified time | relevance | path

Searched refs:excluded_syms (Results 1 – 4 of 4) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/nnvm/
H A Dlow_precision_pass.cc125 const std::unordered_set<std::string> &excluded_syms, ObjectPtr node) { in CheckConditionalFP32() argument
126 if (node->is_variable() || (excluded_syms.count(node->attrs.name) > 0) || in CheckConditionalFP32()
162 const auto excluded_syms = src.GetAttr<std::unordered_set<std::string>>("excluded_syms"); in ReducePrecision() local
203 (excluded_syms.count(node->attrs.name) > 0)) { in ReducePrecision()
225 excluded_syms.count(node->attrs.name) == 0) { in ReducePrecision()
275 excluded_syms.count(node->attrs.name) == 0) { in ReducePrecision()
333 } else if (CheckConditionalFP32(conditional_fp32_ops, excluded_syms, node)) { in ReducePrecision()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/nnvm/
H A Dlow_precision_pass.cc125 const std::unordered_set<std::string> &excluded_syms, ObjectPtr node) { in CheckConditionalFP32() argument
126 if (node->is_variable() || (excluded_syms.count(node->attrs.name) > 0) || in CheckConditionalFP32()
162 const auto excluded_syms = src.GetAttr<std::unordered_set<std::string>>("excluded_syms"); in ReducePrecision() local
203 (excluded_syms.count(node->attrs.name) > 0)) { in ReducePrecision()
225 excluded_syms.count(node->attrs.name) == 0) { in ReducePrecision()
275 excluded_syms.count(node->attrs.name) == 0) { in ReducePrecision()
333 } else if (CheckConditionalFP32(conditional_fp32_ops, excluded_syms, node)) { in ReducePrecision()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/c_api/
H A Dc_api_symbolic.cc1172 std::unordered_set<std::string> excluded_syms; in MXReducePrecisionSymbol() local
1192 excluded_syms.emplace(excluded_symbols[i]); in MXReducePrecisionSymbol()
1222 std::make_shared<nnvm::any>(std::move(excluded_syms)); in MXReducePrecisionSymbol()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/c_api/
H A Dc_api_symbolic.cc1172 std::unordered_set<std::string> excluded_syms; in MXReducePrecisionSymbol() local
1192 excluded_syms.emplace(excluded_symbols[i]); in MXReducePrecisionSymbol()
1222 std::make_shared<nnvm::any>(std::move(excluded_syms)); in MXReducePrecisionSymbol()