Home
last modified time | relevance | path

Searched refs:batch_mean_opaque (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3250 void* batch_mean_opaque; in DoBatchNormalizationForwardImpl() local
3255 batch_mean_opaque = batch_mean->opaque(); in DoBatchNormalizationForwardImpl()
3258 batch_mean_opaque = nullptr; in DoBatchNormalizationForwardImpl()
3265 scale.opaque(), offset.opaque(), 1.0, batch_mean_opaque, in DoBatchNormalizationForwardImpl()