Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1011 miopenStatus_t SetBatchNormForwardArgs(const int op_idx, const float* alpha, in SetBatchNormForwardArgs() function in stream_executor::gpu::ScopedFusionPlanBase
1382 miopenStatus_t SetBatchNormForwardArgs(const void* scale, const void* offset, in SetBatchNormForwardArgs() function in stream_executor::gpu::ScopedFusionPlanBatchNormActivationForward
1388 return ScopedFusionPlanBase::SetBatchNormForwardArgs( in SetBatchNormForwardArgs()
4279 fusion_plan.SetBatchNormForwardArgs( in DoFusedBatchNormActivationForwardImpl()