SetBatchNormInferenceArgs (reference) in projects: external

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

/external/tensorflow/tensorflow/stream_executor/rocm/
D	rocm_dnn.cc	988 miopenStatus_t SetBatchNormInferenceArgs(const int op_idx, const float* alpha, in SetBatchNormInferenceArgs() function in stream_executor::gpu::ScopedFusionPlanBase 1282 miopenStatus_t SetBatchNormInferenceArgs(const void* scale, in SetBatchNormInferenceArgs() function in stream_executor::gpu::ScopedFusionPlanBatchNormActivationInference 1288 return ScopedFusionPlanBase::SetBatchNormInferenceArgs( in SetBatchNormInferenceArgs() 4168 fusion_plan.SetBatchNormInferenceArgs( in DoFusedBatchNormActivationInferenceImpl()