Searched refs:y_before_reduce (Results 1 – 1 of 1) sorted by relevance
469 auto y_before_reduce = grad_output * weight_x * bottom_right_minus_top_right + in CalculateGradWarp() local473 xla::Reduce(y_before_reduce, xla::Zero(ctx->builder(), data_type), in CalculateGradWarp()