Searched refs:reshaped_y (Results 1 – 1 of 1) sorted by relevance
477 XlaOp reshaped_y = xla::Reshape(y_result, reshaped_dims, reshaped_sizes); in CalculateGradWarp() local479 return xla::ConcatInDim(ctx->builder(), {reshaped_x, reshaped_y}, in CalculateGradWarp()