Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc970 miopenStatus_t SetBiasArgs(const int op_idx, const float* alpha, in SetBiasArgs() function in stream_executor::gpu::ScopedFusionPlanBase
1180 miopenStatus_t SetBiasArgs(const void* bias_data) { in SetBiasArgs() function in stream_executor::gpu::ScopedFusionPlanConvolutionBiasActivation
1183 return ScopedFusionPlanBase::SetBiasArgs(k_bias_op_idx, &alpha, &beta, in SetBiasArgs()
4074 status = fusion_plan.SetBiasArgs(bias_data.opaque()); in DoFusedConvolutionBiasActivationImpl()