| /dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/contrib/amp/ |
| H A D | amp.py | 202 wrap_list = fp32_ops if fp32_ops is not None else list_fp32_ops(target_dtype)
283 conditional_fp32_ops=None, fp32_ops=None): argument
315 conditional_fp32_ops, fp32_ops)
317 conditional_fp32_ops, fp32_ops)
374 fp32_ops=None, conditional_fp32_ops=None, argument
423 if fp32_ops is not None:
426 fp32_ops = list_fp32_ops(target_dtype)
456 common_ops = set(target_dtype_ops) & set(fp32_ops)
462 common_ops = set(conditional_op_names) & set(fp32_ops)
591 fp32_ops, conditional_fp32_ops,
[all …]
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/contrib/amp/ |
| H A D | amp.py | 202 wrap_list = fp32_ops if fp32_ops is not None else list_fp32_ops(target_dtype)
283 conditional_fp32_ops=None, fp32_ops=None): argument
315 conditional_fp32_ops, fp32_ops)
317 conditional_fp32_ops, fp32_ops)
374 fp32_ops=None, conditional_fp32_ops=None, argument
423 if fp32_ops is not None:
426 fp32_ops = list_fp32_ops(target_dtype)
456 common_ops = set(target_dtype_ops) & set(fp32_ops)
462 common_ops = set(conditional_op_names) & set(fp32_ops)
591 fp32_ops, conditional_fp32_ops,
[all …]
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/gpu/ |
| H A D | test_contrib_amp.py | 110 fp32_ops=["sin"])
124 fp32_ops=["elemwise_add"])
127 fp32_ops=["Activation"],
131 fp32_ops=["Activation"],
135 fp32_ops=["FullyConnected"])
143 fp32_ops=["sin"],
152 fp32_ops=["sin"],
442 fp32_ops=[], cast_optional_params=True)
453 fp32_ops=[], cast_optional_params=True)
464 fp32_ops=[], cast_optional_params=True)
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/gpu/ |
| H A D | test_contrib_amp.py | 110 fp32_ops=["sin"])
124 fp32_ops=["elemwise_add"])
127 fp32_ops=["Activation"],
131 fp32_ops=["Activation"],
135 fp32_ops=["FullyConnected"])
143 fp32_ops=["sin"],
152 fp32_ops=["sin"],
442 fp32_ops=[], cast_optional_params=True)
453 fp32_ops=[], cast_optional_params=True)
464 fp32_ops=[], cast_optional_params=True)
[all …]
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/mkl/ |
| H A D | test_contrib_amp.py | 109 fp32_ops=["sin"])
122 fp32_ops=["elemwise_add"])
125 fp32_ops=["Activation"],
129 fp32_ops=["Activation"],
133 fp32_ops=["FullyConnected"])
141 fp32_ops=["sin"],
150 fp32_ops=["sin"],
462 fp32_ops=[], cast_optional_params=True)
473 fp32_ops=[], cast_optional_params=True)
484 fp32_ops=[], cast_optional_params=True)
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/mkl/ |
| H A D | test_contrib_amp.py | 109 fp32_ops=["sin"])
122 fp32_ops=["elemwise_add"])
125 fp32_ops=["Activation"],
129 fp32_ops=["Activation"],
133 fp32_ops=["FullyConnected"])
141 fp32_ops=["sin"],
150 fp32_ops=["sin"],
462 fp32_ops=[], cast_optional_params=True)
473 fp32_ops=[], cast_optional_params=True)
484 fp32_ops=[], cast_optional_params=True)
[all …]
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/nnvm/ |
| H A D | low_precision_pass.cc | 157 const auto fp32_ops = in ReducePrecision() local
202 if ((!node->is_variable() && fp32_ops.count(node->op()->name) > 0) || in ReducePrecision()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/nnvm/ |
| H A D | low_precision_pass.cc | 157 const auto fp32_ops = in ReducePrecision() local
202 if ((!node->is_variable() && fp32_ops.count(node->op()->name) > 0) || in ReducePrecision()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/c_api/ |
| H A D | c_api_symbolic.cc | 1170 std::unordered_set<std::string> fp32_ops; in MXReducePrecisionSymbol() local
1186 fp32_ops.emplace(fp32_op_names[i]); in MXReducePrecisionSymbol()
1216 g.attrs["fp32_ops"] = std::make_shared<nnvm::any>(std::move(fp32_ops)); in MXReducePrecisionSymbol()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/c_api/ |
| H A D | c_api_symbolic.cc | 1170 std::unordered_set<std::string> fp32_ops; in MXReducePrecisionSymbol() local
1186 fp32_ops.emplace(fp32_op_names[i]); in MXReducePrecisionSymbol()
1216 g.attrs["fp32_ops"] = std::make_shared<nnvm::any>(std::move(fp32_ops)); in MXReducePrecisionSymbol()
|