Searched refs:bn_ops (Results 1 – 2 of 2) sorted by relevance
| /external/python/cpython3/Mac/BuildScript/ |
| D | openssl-mac-arm64.patch | 5 bn_ops => "SIXTY_FOUR_BIT_LONG",
13 + bn_ops => "SIXTY_FOUR_BIT_LONG",
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1695 const cudnnBatchNormOps_t& bn_ops, in CreateBatchNormForwardWorkspace() argument
1704 /*handle=*/cudnn.handle(), /*mode=*/mode, /*bnOps=*/bn_ops, in CreateBatchNormForwardWorkspace()
1719 const cudnnBatchNormOps_t& bn_ops, in CreateBatchNormBackwardWorkspace() argument
1726 /*handle=*/cudnn.handle(), /*mode=*/mode, /*bnOps=*/bn_ops, in CreateBatchNormBackwardWorkspace()
3532 const cudnnBatchNormOps_t bn_ops = get_bn_ops(); in DoBatchNormalizationForwardImpl() local
3542 stream, cudnn, mode, bn_ops, activation_desc.handle(), x_descriptor, in DoBatchNormalizationForwardImpl()
3548 /*handle=*/cudnn.handle(), /*mode=*/mode, /*bnOps=*/bn_ops, in DoBatchNormalizationForwardImpl()
3596 /*bnOps=*/bn_ops, in DoBatchNormalizationForwardImpl()
3705 const cudnnBatchNormOps_t bn_ops = CUDNN_BATCHNORM_OPS_BN; in DoBatchNormalizationBackwardImpl() local
3708 stream, cudnn, mode, bn_ops, x_descriptor, in DoBatchNormalizationBackwardImpl()
[all …]
|