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