Searched defs:qengine (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/test/quantization/core/ |
| D | test_quantized_functional.py | 91 use_bias, use_channelwise, qengine, argument
143 use_bias, use_channelwise, qengine, argument
200 use_channelwise, qengine, argument
|
| D | test_quantized_op.py | 6394 def test_qconv3d(
6478 def test_qconv3d_relu(
|
| /external/pytorch/c10/core/ |
| D | QEngine.h | 28 inline std::string toString(QEngine qengine) { in toString()
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_quantized.py | 148 def override_quantized_engine(qengine): argument
|
| D | common_quantization.py | 1365 def __init__(self, qengine='fbgemm'): argument
1378 def __init__(self, qengine='fbgemm'): argument
1436 def __init__(self, qengine='fbgemm'): argument
1470 def __init__(self, qengine): argument
1487 def __init__(self, qengine): argument
1553 def __init__(self, qengine='fbgemm'): argument
1878 def __init__(self, qengine): argument
2151 def __init__(self, qengine): argument
2183 def __init__(self, qengine): argument
2200 def __init__(self, qengine): argument
|
| /external/pytorch/aten/src/ATen/ |
| D | Context.cpp | 358 at::QEngine qengine = at::kNoQEngine; in qEngine() local
|
| /external/pytorch/torch/csrc/ |
| D | Module.cpp | 1121 auto qengine = THPUtils_unpackLong(arg); in THPModule_setQEngine() local
|