Searched refs:k_actv_op_idx (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1193 k_actv_op_idx, &alpha, &beta, activation_descriptor.alpha_, in SetActivationForwardArgs()
1224 const int k_actv_op_idx = 2; member in stream_executor::gpu::ScopedFusionPlanConvolutionBiasActivation
1299 k_actv_op_idx, &alpha, &beta, activation_descriptor.alpha_, in SetActivationForwardArgs()
1325 const int k_actv_op_idx = 1; member in stream_executor::gpu::ScopedFusionPlanBatchNormActivationInference
1399 k_actv_op_idx, &alpha, &beta, activation_descriptor.alpha_, in SetActivationForwardArgs()
1425 const int k_actv_op_idx = 1; member in stream_executor::gpu::ScopedFusionPlanBatchNormActivationForward
1501 k_actv_op_idx, &alpha, &beta, y, activation_descriptor.alpha_, in SetActivationBackwardArgs()
1527 const int k_actv_op_idx = 1; member in stream_executor::gpu::ScopedFusionPlanBatchNormActivationBackward
|