Searched refs:weight_x (Results 1 – 1 of 1) sorted by relevance
442 auto weight_x = xla::SliceInDim(ratio, /*start_index=*/0, /*limit_index=*/1, in CalculateGradWarp() local449 auto one_minus_x = xla::One(ctx->builder(), data_type) - weight_x; in CalculateGradWarp()469 auto y_before_reduce = grad_output * weight_x * bottom_right_minus_top_right + in CalculateGradWarp()