Searched defs:FusedBatchNormArgs (Results 1 – 1 of 1) sorted by relevance
147 struct FusedBatchNormArgs { struct148 const T* scale_data = nullptr;149 const T* offset_data = nullptr;150 const T* estimated_mean_data = nullptr;151 const T* estimated_variance_data = nullptr;155 Eigen::Tensor<T, 1, Eigen::RowMajor> scaling_factor;157 float leakyrelu_alpha;159 static bool IsSupported(FusedComputationType fusion) { in IsSupported()