SetBatchNormBackwardArgs (reference) in projects: external

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

/external/tensorflow/tensorflow/stream_executor/rocm/
D	rocm_dnn.cc	1039 miopenStatus_t SetBatchNormBackwardArgs(const int op_idx, const float* alpha, in SetBatchNormBackwardArgs() function in stream_executor::gpu::ScopedFusionPlanBase 1482 miopenStatus_t SetBatchNormBackwardArgs(const void* x, const void* scale, in SetBatchNormBackwardArgs() function in stream_executor::gpu::ScopedFusionPlanBatchNormActivationBackward 1490 return ScopedFusionPlanBase::SetBatchNormBackwardArgs( in SetBatchNormBackwardArgs() 4395 fusion_plan.SetBatchNormBackwardArgs( in DoFusedBatchNormActivationBackwardImpl()