Searched defs:bn_ops (Results 1 – 1 of 1) sorted by relevance
1818 const cudnnBatchNormOps_t& bn_ops, in CreateBatchNormForwardWorkspace()1842 const cudnnBatchNormOps_t& bn_ops, in CreateBatchNormBackwardWorkspace()4643 const cudnnBatchNormOps_t bn_ops = get_bn_ops(); in DoBatchNormalizationForwardImpl() local4816 const cudnnBatchNormOps_t bn_ops = CUDNN_BATCHNORM_OPS_BN; in DoBatchNormalizationBackwardImpl() local