Searched refs:k_batchnorm_op_idx (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1289 k_batchnorm_op_idx, &alpha, &beta, scale, offset, mean, variance, in SetBatchNormInferenceArgs()
1324 const int k_batchnorm_op_idx = 0; member in stream_executor::gpu::ScopedFusionPlanBatchNormActivationInference
1389 k_batchnorm_op_idx, &alpha, &beta, scale, offset, batch_mean, batch_var, in SetBatchNormForwardArgs()
1424 const int k_batchnorm_op_idx = 0; member in stream_executor::gpu::ScopedFusionPlanBatchNormActivationForward
1491 k_batchnorm_op_idx, &alpha, &beta, x, scale, offset, scale_grad, in SetBatchNormBackwardArgs()
1526 const int k_batchnorm_op_idx = 0; member in stream_executor::gpu::ScopedFusionPlanBatchNormActivationBackward
|