Searched defs:bn_ops (Results 1 – 1 of 1) sorted by relevance
1876 const cudnnBatchNormOps_t& bn_ops, in CreateBatchNormForwardWorkspace()1900 const cudnnBatchNormOps_t& bn_ops, in CreateBatchNormBackwardWorkspace()5494 const cudnnBatchNormOps_t bn_ops = get_bn_ops(); in DoBatchNormalizationForwardImpl() local5676 const cudnnBatchNormOps_t bn_ops = [&]() { in DoBatchNormalizationBackwardImpl() local