SetActivationBackwardArgs (reference) in projects: external

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

/external/tensorflow/tensorflow/stream_executor/rocm/
D	rocm_dnn.cc	1084 miopenStatus_t SetActivationBackwardArgs(const int op_idx, const float* alpha, in SetActivationBackwardArgs() function in stream_executor::gpu::ScopedFusionPlanBase 1495 miopenStatus_t SetActivationBackwardArgs( in SetActivationBackwardArgs() function in stream_executor::gpu::ScopedFusionPlanBatchNormActivationBackward 1500 return ScopedFusionPlanBase::SetActivationBackwardArgs( in SetActivationBackwardArgs() 4402 status = fusion_plan.SetActivationBackwardArgs(activation_desc, in DoFusedBatchNormActivationBackwardImpl()