Searched defs:SetBatchNormBackwardArgs (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ | ||
| H A D | rocm_dnn.cc | 1206 miopenStatus_t SetBatchNormBackwardArgs(const int op_idx, const float* alpha, in SetBatchNormBackwardArgs() function in stream_executor::gpu::ScopedFusionPlanBase |