Searched defs:bn_ops (Results 1 – 1 of 1) sorted by relevance
1695 const cudnnBatchNormOps_t& bn_ops, in CreateBatchNormForwardWorkspace()1719 const cudnnBatchNormOps_t& bn_ops, in CreateBatchNormBackwardWorkspace()3532 const cudnnBatchNormOps_t bn_ops = get_bn_ops(); in DoBatchNormalizationForwardImpl() local3705 const cudnnBatchNormOps_t bn_ops = CUDNN_BATCHNORM_OPS_BN; in DoBatchNormalizationBackwardImpl() local