Searched refs:x_before_reduce (Results 1 – 1 of 1) sorted by relevance
451 auto x_before_reduce = in CalculateGradWarp() local463 xla::Reduce(x_before_reduce, xla::Zero(ctx->builder(), data_type), in CalculateGradWarp()