Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc910 class ScopedFusionPlanBase { class
912 ScopedFusionPlanBase(miopenHandle_t miopen_handle, in ScopedFusionPlanBase() function in stream_executor::gpu::ScopedFusionPlanBase
926 virtual ~ScopedFusionPlanBase() { in ~ScopedFusionPlanBase()
1111 SE_DISALLOW_COPY_AND_ASSIGN(ScopedFusionPlanBase);
1115 class ScopedFusionPlanConvolutionBiasActivation : public ScopedFusionPlanBase {
1123 : ScopedFusionPlanBase(miopen_handle, miopenVerticalFusion, in ScopedFusionPlanConvolutionBiasActivation()
1176 return ScopedFusionPlanBase::SetConvolutionArgs(k_conv_op_idx, &alpha, in SetConvolutionArgs()
1183 return ScopedFusionPlanBase::SetBiasArgs(k_bias_op_idx, &alpha, &beta, in SetBiasArgs()
1192 return ScopedFusionPlanBase::SetActivationForwardArgs( in SetActivationForwardArgs()
1231 : public ScopedFusionPlanBase {
[all …]