Searched refs:grad_channels (Results 1 – 1 of 1) sorted by relevance
662 const int64 grad_channels = grad_shape.dim_size(3); in Compile() local671 ctx, channels == grad_channels, in Compile()674 channels, " vs. ", grad_channels, ")")); in Compile()