Searched refs:weight_y (Results 1 – 1 of 1) sorted by relevance
444 auto weight_y = xla::SliceInDim(ratio, /*start_index=*/1, /*limit_index=*/2, in CalculateGradWarp() local448 auto one_minus_y = xla::One(ctx->builder(), data_type) - weight_y; in CalculateGradWarp()452 grad_output * weight_y * bottom_right_minus_bottom_left + in CalculateGradWarp()