SetConvolutionArgs (reference) in projects: external

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

/external/tensorflow/tensorflow/stream_executor/rocm/
D	rocm_dnn.cc	952 miopenStatus_t SetConvolutionArgs(const int op_idx, const float* alpha, in SetConvolutionArgs() function in stream_executor::gpu::ScopedFusionPlanBase 1173 miopenStatus_t SetConvolutionArgs(const void* filter_data) { in SetConvolutionArgs() function in stream_executor::gpu::ScopedFusionPlanConvolutionBiasActivation 1176 return ScopedFusionPlanBase::SetConvolutionArgs(k_conv_op_idx, &alpha, in SetConvolutionArgs() 4070 fusion_plan.SetConvolutionArgs(filter_data.opaque()); in DoFusedConvolutionBiasActivationImpl()