Searched refs:num_output_features (Results 1 – 1 of 1) sorted by relevance
122 int64_t num_output_features = shape.dimensions( in EnsureIsConvBiasActivation() local133 auto bias = BroadcastZeros(comp, bias_ty, {num_output_features}); in EnsureIsConvBiasActivation()