/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()
|