nvmath.bindings.cublasLt.Epilogue¶
- enum nvmath.bindings.cublasLt.Epilogue(value)[source]¶
See
cublasLtEpilogue_t.- Member Type:
Valid values are as follows:
- DEFAULT = <Epilogue.DEFAULT: 1>¶
- RELU = <Epilogue.RELU: 2>¶
- RELU_AUX = <Epilogue.RELU_AUX: 130>¶
- BIAS = <Epilogue.BIAS: 4>¶
- RELU_BIAS = <Epilogue.RELU_BIAS: 6>¶
- RELU_AUX_BIAS = <Epilogue.RELU_AUX_BIAS: 134>¶
- DRELU = <Epilogue.DRELU: 136>¶
- DRELU_BGRAD = <Epilogue.DRELU_BGRAD: 152>¶
- GELU = <Epilogue.GELU: 32>¶
- GELU_AUX = <Epilogue.GELU_AUX: 160>¶
- GELU_BIAS = <Epilogue.GELU_BIAS: 36>¶
- GELU_AUX_BIAS = <Epilogue.GELU_AUX_BIAS: 164>¶
- DGELU = <Epilogue.DGELU: 192>¶
- DGELU_BGRAD = <Epilogue.DGELU_BGRAD: 208>¶
- BGRADA = <Epilogue.BGRADA: 256>¶
- BGRADB = <Epilogue.BGRADB: 512>¶